Merge pull request #238 from Labelbox/jtso/delete_bir

creating a delete method for bulk import requests

Author: msokoloff1

labelbox/schema/bulk_import_request.py

@@ -386,6 +386,21 @@ def create_from_local_file(cls,
                                                       file_name, file_data)
         return cls(client, response_data["createBulkImportRequest"])
 
+    def delete(self) -> None:
+        """ Deletes the import job and also any annotations created by this import.
+        
+        Returns: 
+            None
+        """
+        id_param = "bulk_request_id"
+        query_str = """mutation deleteBulkImportRequestPyApi($%s: ID!) {
+            deleteBulkImportRequest(where: {id: $%s}) {
+                id
+                name
+            }
+        }""" % (id_param, id_param)
+        self.client.execute(query_str, {id_param: self.uid})
+
 
 def _validate_ndjson(lines: Iterable[Dict[str, Any]],
                      project: "labelbox.Project") -> None:

tests/integration/bulk_import/test_bulk_import_request.py

@@ -147,3 +147,23 @@ def test_wait_till_done(rectangle_inference, configured_project):
 def assert_file_content(url: str, predictions):
     response = requests.get(url)
     assert response.text == ndjson.dumps(predictions)
+
+
+def test_delete(client, configured_project, predictions):
+
+    id_param = "project_id"
+    query_str = """query bulk_import_requestsPyApi($%s: ID!) {bulkImportRequests(where: {projectId: $%s}) {id}}""" % (
+        id_param, id_param)
+    name = str(uuid.uuid4())
+
+    bulk_import_request = configured_project.upload_annotations(
+        name=name, annotations=predictions)
+    bulk_import_request.wait_until_done()
+    all_import_requests = client.execute(query_str,
+                                         {id_param: configured_project.uid})
+    assert len(all_import_requests['bulkImportRequests']) == 1
+
+    bulk_import_request.delete()
+    all_import_requests = client.execute(query_str,
+                                         {id_param: configured_project.uid})
+    assert len(all_import_requests['bulkImportRequests']) == 0