From 0a3dcb7d00ac9d349b8ba567f3147cd28d7035f1 Mon Sep 17 00:00:00 2001
From: Dan Davison <dan.davison@temporal.io>
Date: Mon, 1 Jul 2024 10:11:19 -0400
Subject: [PATCH] Fix mypy-specific type check error

---
 tests/worker/test_workflow.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py
index 4b4cbb01..246a83e3 100644
--- 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(