3.8 compat

message_passing/message_handler_waiting_compensation_cleanup/__init__.py

@@ -1,25 +1,25 @@
 from dataclasses import dataclass
-from enum import StrEnum
+from enum import IntEnum
 
 TASK_QUEUE = "my-task-queue"
 WORKFLOW_ID = "my-workflow-id"
 
 
-class WorkflowExitType(StrEnum):
-    SUCCESS = "success"
-    FAILURE = "failure"
-    CONTINUE_AS_NEW = "continue_as_new"
-    CANCELLATION = "cancellation"
+class WorkflowExitType(IntEnum):
+    SUCCESS = 0
+    FAILURE = 1
+    CONTINUE_AS_NEW = 2
+    CANCELLATION = 3
 
 
 @dataclass
 class WorkflowInput:
     exit_type: WorkflowExitType
 
 
-class OnWorkflowExitAction(StrEnum):
-    CONTINUE = "continue"
-    ABORT_WITH_COMPENSATION = "abort_with_compensation"
+class OnWorkflowExitAction(IntEnum):
+    CONTINUE = 0
+    ABORT_WITH_COMPENSATION = 1
 
 
 @dataclass

message_passing/message_handler_waiting_compensation_cleanup/workflows.py

@@ -60,7 +60,7 @@ async def my_update(self, input: UpdateInput) -> str:
         update_task = asyncio.Task(self._my_update())
         # 👉 Always use `workflow.wait` instead of `asyncio.wait` in Workflow
         # code: asyncio's version is non-deterministic.
-        first_completed, _ = await workflow.wait(
+        first_completed, _ = await workflow.wait(  # type: ignore
             [update_task, self.workflow_exit_exception],
             return_when=asyncio.FIRST_COMPLETED,
         )