diff --git a/aixplain/factories/corpus_factory.py b/aixplain/factories/corpus_factory.py index d8cc611e..d14cad10 100644 --- a/aixplain/factories/corpus_factory.py +++ b/aixplain/factories/corpus_factory.py @@ -117,7 +117,7 @@ def get(cls, corpus_id: Text) -> Corpus: Returns: Corpus: Created 'Corpus' object """ - url = urljoin(cls.backend_url, f"sdk/corpus/{corpus_id}/overview") + url = urljoin(cls.backend_url, f"sdk/corpora/{corpus_id}/overview") if cls.aixplain_key != "": headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"} else: @@ -163,7 +163,7 @@ def list( Returns: Dict: list of corpora in agreement with the filters, page number, page total and total elements """ - url = urljoin(cls.backend_url, "sdk/corpus/paginate") + url = urljoin(cls.backend_url, "sdk/corpora/paginate") if cls.aixplain_key != "": headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"} else: diff --git a/aixplain/factories/dataset_factory.py b/aixplain/factories/dataset_factory.py index 54d98ace..f1c8a74e 100644 --- a/aixplain/factories/dataset_factory.py +++ b/aixplain/factories/dataset_factory.py @@ -165,7 +165,7 @@ def get(cls, dataset_id: Text) -> Dataset: Returns: Dataset: Created 'Dataset' object """ - url = urljoin(cls.backend_url, f"sdk/dataset/{dataset_id}/overview") + url = urljoin(cls.backend_url, f"sdk/datasets/{dataset_id}/overview") if cls.aixplain_key != "": headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"} else: @@ -206,7 +206,7 @@ def list( Returns: Dict: list of datasets in agreement with the filters, page number, page total and total elements """ - url = urljoin(cls.backend_url, "sdk/dataset/paginate") + url = urljoin(cls.backend_url, "sdk/datasets/paginate") if cls.aixplain_key != "": headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"} else: diff --git a/setup.py b/setup.py index ba752325..b8a7f591 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ sys.exit(1) requires = ["requests>=2.1.0", "tqdm>=4.1.0", "pandas>=1.2.1", "python-dotenv>=1.0.0", "validators>=0.20.0", "filetype>=1.2.0"] -test_requirements = ["python-dotenv~=0.20.0", "pytest>=6.1.0"] +test_requirements = ["python-dotenv~=0.20.0", "coverage>=7.3.2", "pytest>=6.1.0", "requests-mock>=1.11.0"] about = {} here = os.path.abspath(os.path.dirname(__file__))