Add type annotation needed by mypy

temporalio · Jul 15, 2024 · 9266c22 · 9266c22
1 parent 33d7701
commit 9266c22
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py
@@ -5735,7 +5735,10 @@ async def test_workflow_return_is_honored_when_it_precedes_signal_completion_com
 async def _do_first_completion_command_is_honored_test(
     client: Client, main_workflow_returns_before_signal_completions: bool
 ):
-    workflow_cls = (
+    workflow_cls: Union[
+        Type[FirstCompletionCommandIsHonoredPingPongWorkflow],
+        Type[FirstCompletionCommandIsHonoredWorkflow],
+    ] = (
         FirstCompletionCommandIsHonoredPingPongWorkflow
         if main_workflow_returns_before_signal_completions
         else FirstCompletionCommandIsHonoredWorkflow