File tree Expand file tree Collapse file tree 2 files changed +9
-6
lines changed
packages/examples/cvat/exchange-oracle/src/endpoints Expand file tree Collapse file tree 2 files changed +9
-6
lines changed Original file line number Diff line number Diff line change 2828from src .endpoints .serializers import (
2929 ASSIGNMENT_PROJECT_VALIDATION_STATUSES ,
3030 PROJECT_ACTIVE_STATUSES ,
31+ PROJECT_COMPLETED_STATUSES ,
3132 serialize_assignment ,
3233 serialize_job ,
3334)
@@ -146,9 +147,7 @@ async def list_jobs(
146147 cvat_service .Project .status == cvat_service .ProjectStatuses .canceled
147148 )
148149 case JobStatuses .completed :
149- query = query .filter (
150- cvat_service .Project .status == cvat_service .ProjectStatuses .recorded
151- )
150+ query = query .filter (cvat_service .Project .status .in_ (PROJECT_COMPLETED_STATUSES ))
152151 case _:
153152 raise NotImplementedError (f"Unsupported status { status } " )
154153
Original file line number Diff line number Diff line change 1212from src .schemas import exchange as service_api
1313from src .utils .assignments import compose_assignment_url , parse_manifest
1414
15- PROJECT_ACTIVE_STATUSES = [
15+ PROJECT_ACTIVE_STATUSES = {
1616 ProjectStatuses .annotation ,
1717 ProjectStatuses .completed ,
1818 ProjectStatuses .validation ,
19- ]
19+ }
20+ PROJECT_COMPLETED_STATUSES = {
21+ ProjectStatuses .recorded ,
22+ ProjectStatuses .deleted ,
23+ }
2024
2125
2226def serialize_job (
@@ -48,7 +52,7 @@ def serialize_job(
4852 api_status = service_api .JobStatuses .canceled
4953 elif project .status in PROJECT_ACTIVE_STATUSES :
5054 api_status = service_api .JobStatuses .active
51- elif project .status == ProjectStatuses . recorded :
55+ elif project .status in PROJECT_COMPLETED_STATUSES :
5256 api_status = service_api .JobStatuses .completed
5357 else :
5458 raise NotImplementedError (f"Unknown status { project .status } " )
You can’t perform that action at this time.
0 commit comments