fix: missing retried python (#426)

aneoconsulting · Oct 10, 2023 · 9ce98b2 · 9ce98b2
2 parents ca5b236 + 661cb3c
commit 9ce98b2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/python/src/armonik/common/enumwrapper.py b/packages/python/src/armonik/common/enumwrapper.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from ..protogen.common.task_status_pb2 import TaskStatus as RawStatus, _TASKSTATUS, TASK_STATUS_CANCELLED, TASK_STATUS_CANCELLING, TASK_STATUS_COMPLETED, TASK_STATUS_CREATING, TASK_STATUS_DISPATCHED, TASK_STATUS_ERROR, TASK_STATUS_PROCESSED, TASK_STATUS_PROCESSING, TASK_STATUS_SUBMITTED, TASK_STATUS_TIMEOUT, TASK_STATUS_UNSPECIFIED
+from ..protogen.common.task_status_pb2 import TaskStatus as RawStatus, _TASKSTATUS, TASK_STATUS_CANCELLED, TASK_STATUS_CANCELLING, TASK_STATUS_COMPLETED, TASK_STATUS_CREATING, TASK_STATUS_DISPATCHED, TASK_STATUS_ERROR, TASK_STATUS_PROCESSED, TASK_STATUS_PROCESSING, TASK_STATUS_SUBMITTED, TASK_STATUS_TIMEOUT, TASK_STATUS_UNSPECIFIED, TASK_STATUS_RETRIED
 from ..protogen.common.worker_common_pb2 import HealthCheckReply
 from ..protogen.common.sort_direction_pb2 import SORT_DIRECTION_ASC, SORT_DIRECTION_DESC
 
@@ -27,6 +27,7 @@ def name_from_value(status: RawStatus) -> str:
     PROCESSED = TASK_STATUS_PROCESSED
     PROCESSING = TASK_STATUS_PROCESSING
     SUBMITTED = TASK_STATUS_SUBMITTED
+    RETRIED = TASK_STATUS_RETRIED
     TIMEOUT = TASK_STATUS_TIMEOUT
     UNSPECIFIED = TASK_STATUS_UNSPECIFIED