creating a delete method for bulk import requests

Author: jtsodapop

labelbox/schema/bulk_import_request.py

@@ -11,8 +11,8 @@
 import requests
 from pydantic import BaseModel, validator
 from typing_extensions import Literal
-from typing import (Any, List, Optional, BinaryIO, Dict, Iterable, Tuple, Union,
-                    Type, Set)
+from typing import (Any, List, Optional, BinaryIO, Dict, Iterable, Tuple,
+                    Union, Type, Set)
 
 import labelbox
 from labelbox import utils
@@ -79,13 +79,14 @@ def _send_create_file_command(
     response_data = response_json.get("data", None)
     if response_data is None:
         raise labelbox.exceptions.LabelboxError(
-            "Failed to upload, message: %s" % response_json.get("errors", None))
+            "Failed to upload, message: %s" %
+            response_json.get("errors", None))
 
     if not response_data.get("createBulkImportRequest", None):
         raise labelbox.exceptions.LabelboxError(
             "Failed to create BulkImportRequest, message: %s" %
-            response_json.get("errors", None) or
-            response_data.get("error", None))
+            response_json.get("errors", None)
+            or response_data.get("error", None))
 
     return response_data
 
@@ -386,6 +387,22 @@ def create_from_local_file(cls,
                                                       file_name, file_data)
         return cls(client, response_data["createBulkImportRequest"])
 
+    def delete(self) -> None:
+        """ Deletes the BulkImportRequest object
+        
+        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:
@@ -418,7 +435,8 @@ def _validate_ndjson(lines: Iterable[Dict[str, Any]],
                     f'{uuid} already used in this import job, '
                     'must be unique for the project.')
             uids.add(uuid)
-        except (pydantic.ValidationError, ValueError, TypeError, KeyError) as e:
+        except (pydantic.ValidationError, ValueError, TypeError,
+                KeyError) as e:
             raise labelbox.exceptions.MALValidationError(
                 f"Invalid NDJson on line {idx}") from e
 
@@ -502,7 +520,6 @@ class VideoSupported(BaseModel):
 #Base class for a special kind of union.
 # Compatible with pydantic. Improves error messages over a traditional union
 class SpecialUnion:
-
     def __new__(cls, **kwargs):
         return cls.build(kwargs)
 
@@ -640,10 +657,11 @@ def validate_answers(cls, value, field):
 
     def validate_feature_schemas(self, valid_feature_schemas):
         #Test top level feature schema for this tool
-        super(NDChecklist, self).validate_feature_schemas(valid_feature_schemas)
+        super(NDChecklist,
+              self).validate_feature_schemas(valid_feature_schemas)
         #Test the feature schemas provided to the answer field
-        if len(set([answer.schemaId for answer in self.answers])) != len(
-                self.answers):
+        if len(set([answer.schemaId
+                    for answer in self.answers])) != len(self.answers):
             raise ValueError(
                 f"Duplicated featureSchema found for checklist {self.uuid}")
         for answer in self.answers:
@@ -712,7 +730,8 @@ class NDPolygon(NDBaseTool):
     def is_geom_valid(cls, v):
         if len(v) < 3:
             raise ValueError(
-                f"A polygon must have at least 3 points to be valid. Found {v}")
+                f"A polygon must have at least 3 points to be valid. Found {v}"
+            )
         return v
 
 
@@ -784,7 +803,8 @@ def is_valid_mask(cls, v):
         #Does the dtype matter? Can it be a float?
         if not isinstance(colors, (tuple, list)):
             raise ValueError(
-                f"Received color that is not a list or tuple. Found : {colors}")
+                f"Received color that is not a list or tuple. Found : {colors}"
+            )
         elif len(colors) != 3:
             raise ValueError(
                 f"Must provide RGB values for segmentation colors. Found : {colors}"
@@ -800,15 +820,14 @@ class NDTool(
         SpecialUnion,
         Type[Union[NDMask,  # type: ignore
                    NDTextEntity, NDPoint, NDRectangle, NDPolyline,
-                   NDPolygon,]]):
+                   NDPolygon, ]]):
     ...
 
 
 class NDAnnotation(
         SpecialUnion,
         Type[Union[NDTool,  # type: ignore
                    NDClassification]]):
-
     @classmethod
     def build(cls: Any, data) -> "NDBase":
         if not isinstance(data, dict):