add job status key constant and fix return for delete_annotations

Author: Bihan Jiang

nucleus/__init__.py

@@ -87,7 +87,7 @@
     ERROR_PAYLOAD,
     ERRORS_KEY,
     JOB_ID_KEY,
-    JOB_STATUS_KEY,
+    JOB_LAST_KNOWN_STATUS_KEY,
     JOB_TYPE_KEY,
     JOB_CREATION_TIME_KEY,
     IMAGE_KEY,
@@ -217,7 +217,7 @@ def list_jobs(
         return [
             AsyncJob(
                 job_id=job[JOB_ID_KEY],
-                job_last_known_status=job[JOB_STATUS_KEY],
+                job_last_known_status=job[JOB_LAST_KNOWN_STATUS_KEY],
                 job_type=job[JOB_TYPE_KEY],
                 job_creation_time=job[JOB_CREATION_TIME_KEY],
                 client=self,

nucleus/constants.py

@@ -42,7 +42,8 @@
 ITEM_METADATA_SCHEMA_KEY = "item_metadata_schema"
 JOB_ID_KEY = "job_id"
 KEEP_HISTORY_KEY = "keep_history"
-JOB_STATUS_KEY = "job_last_known_status"
+JOB_STATUS_KEY = "job_status"
+JOB_LAST_KNOWN_STATUS_KEY = "job_last_known_status"
 JOB_TYPE_KEY = "job_type"
 JOB_CREATION_TIME_KEY = "job_creation_time"
 LABEL_KEY = "label"

nucleus/dataset.py

@@ -366,4 +366,4 @@ def delete_annotations(
         response = self._client.delete_annotations(
             self.id, reference_ids, keep_history
         )
-        return AsyncJob(response[JOB_ID_KEY], self._client)
+        return AsyncJob.from_json(response, self._client)

nucleus/job.py

@@ -6,6 +6,7 @@
     JOB_CREATION_TIME_KEY,
     JOB_ID_KEY,
     JOB_STATUS_KEY,
+    JOB_LAST_KNOWN_STATUS_KEY,
     JOB_TYPE_KEY,
 )
 
@@ -26,7 +27,7 @@ def status(self) -> Dict[str, str]:
             route=f"job/{self.job_id}",
             requests_command=requests.get,
         )
-        self.job_last_known_status = response["job_status"]
+        self.job_last_known_status = response[JOB_STATUS_KEY]
         return response
 
     def errors(self) -> List[str]:
@@ -56,7 +57,7 @@ def sleep_until_complete(self, verbose_std_out=True):
     def from_json(cls, payload: dict, client):
         return cls(
             job_id=payload[JOB_ID_KEY],
-            job_last_known_status=payload[JOB_STATUS_KEY],
+            job_last_known_status=payload[JOB_LAST_KNOWN_STATUS_KEY],
             job_type=payload[JOB_TYPE_KEY],
             job_creation_time=payload[JOB_CREATION_TIME_KEY],
             client=client,