Refactoring: replace Task.status_type with Task.status_as_enum

Author: paulnoirel

libs/labelbox/src/labelbox/client.py

@@ -2442,7 +2442,7 @@ def cancel_task(self, task_id: str) -> bool:
             )
 
         # Check if task is in progress
-        if task.status_type != TaskStatus.In_Progress:
+        if task.status_as_enum != TaskStatus.In_Progress:
             raise LabelboxError(
                 f"Task cannot be cancelled because it is not in progress. Current status: {task.status}"
             )

libs/labelbox/src/labelbox/schema/task.py

@@ -46,8 +46,8 @@ class Task(DbObject):
     created_at = Field.DateTime("created_at")
     name = Field.String("name")
     status = Field.String("status")
-    status_type = Field.Enum(
-        TaskStatus, "status_type", "status"
+    status_as_enum = Field.Enum(
+        TaskStatus, "status_as_enum", "status"
     )  # additional status for filtering
     completion_percentage = Field.Float("completion_percentage")
     result_url = Field.String("result_url", "result")

libs/labelbox/tests/data/export/streamable/test_export_data_rows_streamable.py

@@ -148,7 +148,7 @@ def test_task_filter(self, client, data_row, wait_for_data_row_processing):
 
         # Check if task is listed "in progress" in organization's tasks
         org_tasks_in_progress = organization.tasks(
-            where=Task.status_type == TaskStatus.In_Progress
+            where=Task.status_as_enum == TaskStatus.In_Progress
         )
         retrieved_task_in_progress = next(
             (t for t in org_tasks_in_progress if t.uid == export_task.uid), ""
@@ -159,7 +159,7 @@ def test_task_filter(self, client, data_row, wait_for_data_row_processing):
 
         # Check if task is listed "complete" in user's created tasks
         user_tasks_complete = user.created_tasks(
-            where=Task.status_type == TaskStatus.Complete
+            where=Task.status_as_enum == TaskStatus.Complete
         )
         retrieved_task_complete = next(
             (t for t in user_tasks_complete if t.uid == export_task.uid), ""