Fix mypy-specific type check error

temporalio · Aug 19, 2024 · 65995d5 · 65995d5
1 parent bbb6964
commit 65995d5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py
@@ -5909,9 +5909,10 @@ def init(self, params: UseLockOrSemaphoreWorkflowParameters):
     @workflow.run
     async def run(
         self,
-        params: UseLockOrSemaphoreWorkflowParameters,
+        params: Optional[UseLockOrSemaphoreWorkflowParameters],
     ) -> LockOrSemaphoreWorkflowConcurrencySummary:
         # TODO: Use workflow init method when it exists.
+        assert params
         self.init(params)
         await asyncio.gather(
             *(self.coroutine(f"{i}") for i in range(self.params.n_coroutines))
@@ -5966,6 +5967,7 @@ def __init__(self) -> None:
     @workflow.run
     async def run(
         self,
+        _: Optional[UseLockOrSemaphoreWorkflowParameters] = None,
     ) -> LockOrSemaphoreWorkflowConcurrencySummary:
         await workflow.wait_condition(lambda: self.workflow_may_exit)
         return LockOrSemaphoreWorkflowConcurrencySummary(