Update cancel_task to add clear unique_id option

Author: artliou-scale

scaleapi/__init__.py

@@ -64,18 +64,22 @@ def get_task(self, task_id: str) -> Task:
         endpoint = f"task/{task_id}"
         return Task(self.api.get_request(endpoint), self)
 
-    def cancel_task(self, task_id: str) -> Task:
+    def cancel_task(self, task_id: str, clear_unique_id=False) -> Task:
         """Cancels a task and returns the associated task.
         Raises a ScaleException if it has already been canceled.
 
         Args:
             task_id (str):
                 Task id
-
+            clear_unique_id (boolean):
+                Option to clear unique id when the task is deleted
         Returns:
             Task
         """
-        endpoint = f"task/{task_id}/cancel"
+        if clear_unique_id:
+            endpoint = f"task/{task_id}/cancel?clear_unique_id=true"
+        else:
+            endpoint = f"task/{task_id}/cancel"
         return Task(self.api.post_request(endpoint), self)
 
     def tasks(self, **kwargs) -> Tasklist:
@@ -792,11 +796,7 @@ def import_file(self, file_url: str, **kwargs) -> File:
         filedata = self.api.post_request(endpoint, body=payload)
         return File(filedata, self)
 
-    def create_evaluation_task(
-        self,
-        task_type: TaskType,
-        **kwargs,
-    ) -> EvaluationTask:
+    def create_evaluation_task(self, task_type: TaskType, **kwargs,) -> EvaluationTask:
         """This method can only be used for Rapid projects.
         Supported Task Types: [
             DocumentTranscription,
@@ -833,11 +833,7 @@ def create_evaluation_task(
         evaluation_task_data = self.api.post_request(endpoint, body=kwargs)
         return EvaluationTask(evaluation_task_data, self)
 
-    def create_training_task(
-        self,
-        task_type: TaskType,
-        **kwargs,
-    ) -> TrainingTask:
+    def create_training_task(self, task_type: TaskType, **kwargs,) -> TrainingTask:
         """This method can only be used for Rapid projects.
         Supported Task Types: [
             DocumentTranscription,

tests/test_client.py

@@ -390,8 +390,7 @@ def test_get_batches():
 def test_files_upload():
     with open("tests/test_image.png", "rb") as f:
         client.upload_file(
-            file=f,
-            project_name=TEST_PROJECT_NAME,
+            file=f, project_name=TEST_PROJECT_NAME,
         )