initial commit

Author: sasha-scale

nucleus/__init__.py

@@ -89,6 +89,7 @@
     IMAGE_URL_KEY,
     ITEM_METADATA_SCHEMA_KEY,
     ITEMS_KEY,
+    KEEP_HISTORY_KEY,
     MODEL_RUN_ID_KEY,
     NAME_KEY,
     NUCLEUS_ENDPOINT,
@@ -1036,6 +1037,28 @@ def delete_slice(self, slice_id: str) -> dict:
         )
         return response
 
+    def delete_annotations(
+        self, dataset_id: str, reference_ids: list = None, keep_history=False
+    ) -> dict:
+        """
+        This endpoint deletes annotations.
+
+        :param
+        slice_id: id of the slice
+
+        :return:
+        {}
+        """
+        payload = {KEEP_HISTORY_KEY: keep_history}
+        if reference_ids:
+            payload[REFERENCE_IDS_KEY] = reference_ids
+        response = self.make_request(
+            {},
+            f"annotation/{dataset_id}",
+            requests_command=requests.delete,
+        )
+        return response
+
     def append_to_slice(
         self,
         slice_id: str,

nucleus/constants.py

@@ -40,6 +40,7 @@
 ITEM_KEY = "item"
 ITEM_METADATA_SCHEMA_KEY = "item_metadata_schema"
 JOB_ID_KEY = "job_id"
+KEEP_HISTORY_KEY = "keep_history"
 LABEL_KEY = "label"
 MASK_URL_KEY = "mask_url"
 MESSAGE_KEY = "message"

nucleus/dataset.py

@@ -357,3 +357,10 @@ def items_and_annotations(
             requests_command=requests.get,
         )
         return convert_export_payload(api_payload[EXPORTED_ROWS])
+
+    def delete_annotations(
+        self, reference_ids: list = None, keep_history=False
+    ):
+        return self._client.delete_annotations(
+            self.id, reference_ids, keep_history
+        )