Skip to content

Commit 5de7f9e

Browse files
committed
Update cvat task creation status checks
1 parent 56240e4 commit 5de7f9e

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

packages/examples/cvat/exchange-oracle/src/cvat/api_calls.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -589,8 +589,17 @@ def get_task_upload_status(cvat_id: int) -> tuple[UploadStatus | None, str]:
589589

590590
with get_api_client() as api_client:
591591
try:
592-
(status, _) = api_client.tasks_api.retrieve_status(cvat_id)
593-
return UploadStatus(status.state.value), status.message
592+
results = api_client.requests_api.list(task_id=cvat_id, action="create", page_size=1)[
593+
0
594+
].results
595+
if not results:
596+
status = None
597+
reason = f"Task #{cvat_id} creation request not found"
598+
else:
599+
status = UploadStatus(results[0].status.value.capitalize())
600+
reason = results[0].message
601+
602+
return status, reason
594603
except exceptions.ApiException as e:
595604
if e.status == 404:
596605
return None, e.body

0 commit comments

Comments
 (0)