Merge pull request #122 from scaleapi/jihan/bugfix

Fix autotag export api endpoint path

Author: ardila

nucleus/dataset.py

@@ -110,7 +110,7 @@ def autotag_items(self, autotag_name, for_scores_greater_than=0):
         """
         response = self._client.make_request(
             payload={AUTOTAG_SCORE_THRESHOLD: for_scores_greater_than},
-            route=f"autotag/dataset/{self.id}/autotag/{autotag_name}/taggedItems",
+            route=f"dataset/{self.id}/autotag/{autotag_name}/taggedItems",
             requests_command=requests.get,
         )
         return response
@@ -136,7 +136,7 @@ def autotag_training_items(self, autotag_name):
         """
         response = self._client.make_request(
             payload={},
-            route=f"autotag/dataset/{self.id}/autotag/{autotag_name}/trainingItems",
+            route=f"dataset/{self.id}/autotag/{autotag_name}/trainingItems",
             requests_command=requests.get,
         )
         return response

pyproject.toml

@@ -21,7 +21,7 @@ exclude = '''
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.1.20"
+version = "0.1.21"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]

tests/test_dataset.py

@@ -332,11 +332,11 @@ def test_raises_error_for_duplicate():
 def test_dataset_export_autotag_tagged_items(CLIENT):
     # This test can only run for the test user who has an indexed dataset.
     # TODO: if/when we can create autotags via api, create one instead.
-    if NUCLEUS_PYTEST_USER_ID in CLIENT.api_key:
+    if NUCLEUS_PYTEST_USER_ID in os.environ["NUCLEUS_PYTEST_USER_ID"]:
         dataset = CLIENT.get_dataset(DATASET_WITH_AUTOTAG)
 
         with pytest.raises(NucleusAPIError) as api_error:
-            dataset.autotag_scores(autotag_name="NONSENSE_GARBAGE")
+            dataset.autotag_items(autotag_name="NONSENSE_GARBAGE")
         assert (
             f"The autotag NONSENSE_GARBAGE was not found in dataset {DATASET_WITH_AUTOTAG}"
             in str(api_error.value)
@@ -362,19 +362,19 @@ def test_dataset_export_autotag_tagged_items(CLIENT):
 def test_dataset_export_autotag_training_items(CLIENT):
     # This test can only run for the test user who has an indexed dataset.
     # TODO: if/when we can create autotags via api, create one instead.
-    if NUCLEUS_PYTEST_USER_ID in CLIENT.api_key:
+    if NUCLEUS_PYTEST_USER_ID in os.environ["NUCLEUS_PYTEST_USER_ID"]:
         dataset = CLIENT.get_dataset(DATASET_WITH_AUTOTAG)
 
         with pytest.raises(NucleusAPIError) as api_error:
-            dataset.autotag_scores(autotag_name="NONSENSE_GARBAGE")
+            dataset.autotag_training_items(autotag_name="NONSENSE_GARBAGE")
         assert (
             f"The autotag NONSENSE_GARBAGE was not found in dataset {DATASET_WITH_AUTOTAG}"
             in str(api_error.value)
         )
 
         items = dataset.autotag_training_items(autotag_name="PytestTestTag")
 
-        assert "autotagItems" in items
+        assert "autotagPositiveTrainingItems" in items
         assert "autotag" in items
 
         autotagTrainingItems = items["autotagPositiveTrainingItems"]