Skip to content

Commit 390ee94

Browse files
committed
Fix mypy-specific type check error
1 parent a4d39c9 commit 390ee94

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/worker/test_workflow.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5558,9 +5558,10 @@ def init(self, params: UseLockOrSemaphoreWorkflowParameters):
55585558
@workflow.run
55595559
async def run(
55605560
self,
5561-
params: UseLockOrSemaphoreWorkflowParameters,
5561+
params: Optional[UseLockOrSemaphoreWorkflowParameters],
55625562
) -> LockOrSemaphoreWorkflowConcurrencySummary:
55635563
# TODO: Use workflow init method when it exists.
5564+
assert params
55645565
self.init(params)
55655566
await asyncio.gather(
55665567
*(self.coroutine(f"{i}") for i in range(self.params.n_coroutines))
@@ -5615,6 +5616,7 @@ def __init__(self) -> None:
56155616
@workflow.run
56165617
async def run(
56175618
self,
5619+
_: Optional[UseLockOrSemaphoreWorkflowParameters] = None,
56185620
) -> LockOrSemaphoreWorkflowConcurrencySummary:
56195621
await workflow.wait_condition(lambda: self.workflow_may_exit)
56205622
return LockOrSemaphoreWorkflowConcurrencySummary(

0 commit comments

Comments
 (0)