add support for listing autotags

Author: sasha-scale

nucleus/__init__.py

@@ -103,6 +103,7 @@
     DATASET_LENGTH_KEY,
     NAME_KEY,
     ANNOTATIONS_KEY,
+    AUTOTAGS_KEY,
 )
 from .model import Model
 from .errors import (
@@ -838,6 +839,19 @@ def delete_slice(self, slice_id: str) -> dict:
         )
         return response
 
+    def list_autotags(self, dataset_id: str) -> List[str]:
+        """
+        Fetches a list of autotags for a given dataset id
+        :param dataset_id: internally controlled dataset_id
+        :return: List[str] representing autotag_ids
+        """
+        response = self._make_request(
+            {},
+            f"{dataset_id}/list_autotags",
+            requests_command=requests.get,
+        )
+        return response[AUTOTAGS_KEY] if AUTOTAGS_KEY in response else response
+
     def _make_grequest(
         self,
         payload: dict,

nucleus/constants.py

@@ -1,4 +1,5 @@
-NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
+# NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
+NUCLEUS_ENDPOINT = "http://localhost:3000/v1/nucleus"
 ITEMS_KEY = "items"
 ITEM_KEY = "item"
 REFERENCE_ID_KEY = "reference_id"
@@ -42,3 +43,4 @@
 BOX_TYPE = "box"
 POLYGON_TYPE = "polygon"
 GEOMETRY_KEY = "geometry"
+AUTOTAGS_KEY = "autotags"

nucleus/dataset.py

@@ -221,6 +221,9 @@ def delete_item(self, item_id: str = None, reference_id: str = None):
             self.id, reference_id=reference_id, item_id=item_id
         )
 
+    def list_autotags(self):
+        return self._client.list_autotags(self.id)
+
     def _format_dataset_item_response(self, response: dict) -> dict:
         item = response.get(ITEM_KEY, None)
         annotation_payload = response.get(ANNOTATIONS_KEY, [])

nucleus/slice.py

@@ -1,5 +1,9 @@
 from typing import List, Dict, Any
-from .constants import DATASET_ITEM_ID_KEY, REFERENCE_IDS_KEY
+from .constants import (
+    DATASET_ITEM_ID_KEY,
+    REFERENCE_IDS_KEY,
+    DATASET_ITEM_IDS_KEY,
+)
 
 
 class Slice:
@@ -29,9 +33,9 @@ def info(self, id_type: str = DATASET_ITEM_ID_KEY) -> dict:
         return self._client.slice_info(self.slice_id, id_type)
 
     def append(
-        self, 
+        self,
         dataset_item_ids: List[str] = None,
-        reference_ids: List[str] = None
+        reference_ids: List[str] = None,
     ) -> dict:
         """
         Appends to a slice from items already present in a dataset.
@@ -61,4 +65,4 @@ def append(
         response = self._client._make_request(
             payload, f"slice/{self.slice_id}/append"
         )
-        return response
+        return response