Fix mypy-specific type check error

temporalio · Jul 1, 2024 · 6eb176a · 6eb176a
1 parent a14cfbe
commit 6eb176a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py
@@ -5553,9 +5553,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))
@@ -5610,6 +5611,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(