Merge pull request #70 from scaleapi/da/send_to_labeling

added call to endpoint

Author: ardila

nucleus/slice.py

@@ -2,6 +2,7 @@
 from nucleus.dataset_item import DatasetItem
 from nucleus.annotation import Annotation
 from nucleus.utils import format_dataset_item_response
+from nucleus.job import AsyncJob
 
 from .constants import DEFAULT_ANNOTATION_UPDATE_MODE
 
@@ -145,6 +146,12 @@ def annotate(
             batch_size=batch_size,
         )
 
+    def send_to_labeling(self, project_id: str):
+        response = self._client.make_request(
+            {}, f"slice/{self.slice_id}/{project_id}/send_to_labeling"
+        )
+        return AsyncJob(response["job_id"], self._client)
+
 
 def check_annotations_are_in_slice(
     annotations: List[Annotation], slice_to_check: Slice

tests/helpers.py

@@ -10,6 +10,7 @@
 TEST_MODEL_RUN = "[PyTest] Test Model Run"
 TEST_DATASET_NAME = "[PyTest] Test Dataset"
 TEST_SLICE_NAME = "[PyTest] Test Slice"
+TEST_PROJECT_ID = "60b699d70f139e002dd31bfc"
 
 
 TEST_IMG_URLS = [

tests/test_slice.py

@@ -6,8 +6,10 @@
     TEST_IMG_URLS,
     TEST_SLICE_NAME,
     TEST_BOX_ANNOTATIONS,
+    TEST_PROJECT_ID,
     reference_id_from_url,
 )
+from nucleus.job import AsyncJob
 
 
 @pytest.fixture()
@@ -141,3 +143,29 @@ def sort_by_reference_id(items):
     assert sort_by_reference_id(all_stored_items) == sort_by_reference_id(
         ds_items[:3]
     )
+
+
+def test_slice_send_to_labeling(dataset):
+    # Dataset upload
+    ds_items = []
+    for url in TEST_IMG_URLS:
+        ds_items.append(
+            DatasetItem(
+                image_location=url,
+                reference_id=reference_id_from_url(url),
+            )
+        )
+    response = dataset.append(ds_items)
+    assert ERROR_PAYLOAD not in response.json()
+
+    # Slice creation
+    slc = dataset.create_slice(
+        name=TEST_SLICE_NAME,
+        reference_ids=[ds_items[0].reference_id, ds_items[1].reference_id],
+    )
+
+    response = slc.info()
+    assert len(response["dataset_items"]) == 2
+
+    response = slc.send_to_labeling(TEST_PROJECT_ID)
+    assert isinstance(response, AsyncJob)