Refactored

Author: Claire Pajot

nucleus/__init__.py

@@ -128,7 +128,6 @@
     construct_box_predictions_payload,
     construct_model_creation_payload,
     construct_segmentation_payload,
-    construct_taxonomy_payload,
 )
 from .prediction import (
     BoxPrediction,
@@ -1250,27 +1249,6 @@ def create_image_index(self, dataset_id: str):
             requests_command=requests.post,
         )
 
-    def add_taxonomy(
-        self,
-        dataset_id: str,
-        taxonomy_name: str,
-        taxonomy_type: str,
-        labels: List[str],
-    ):
-        """
-        Creates a new taxonomy.
-        Returns a response with dataset_id, taxonomy_name and type for the new taxonomy.
-        :param dataset_id: id of dataset to add the taxonomy to
-        :param taxonomy_name: name of the taxonomy
-        :param type: type of the taxonomy
-        :param labels: list of possible labels for the taxonomy
-        """
-        return self.make_request(
-            construct_taxonomy_payload(taxonomy_name, taxonomy_type, labels),
-            f"dataset/{dataset_id}/add_taxonomy",
-            requests_command=requests.post,
-        )
-
     def make_request(
         self, payload: dict, route: str, requests_command=requests.post
     ) -> dict:

nucleus/dataset.py

@@ -36,6 +36,7 @@
 from .payload_constructor import (
     construct_append_scenes_payload,
     construct_model_run_creation_payload,
+    construct_taxonomy_payload,
 )
 
 WARN_FOR_LARGE_UPLOAD = 50000
@@ -408,10 +409,22 @@ def create_image_index(self):
         return AsyncJob.from_json(response, self._client)
 
     def add_taxonomy(
-        self, taxonomy_name: str, taxonomy_type: str, labels: List[str]
+        self,
+        taxonomy_name: str,
+        taxonomy_type: str,
+        labels: List[str],
     ):
-        return self._client.add_taxonomy(
-            self.id, taxonomy_name, taxonomy_type, labels
+        """
+        Creates a new taxonomy.
+        Returns a response with dataset_id, taxonomy_name and type for the new taxonomy.
+        :param taxonomy_name: name of the taxonomy
+        :param type: type of the taxonomy
+        :param labels: list of possible labels for the taxonomy
+        """
+        return self._client.make_request(
+            construct_taxonomy_payload(taxonomy_name, taxonomy_type, labels),
+            f"dataset/{self.id}/add_taxonomy",
+            requests_command=requests.post,
         )
 
     def check_index_status(self, job_id: str):