Skip to content

Commit c728e2d

Browse files
committed
Fix mypy-specific type check error
1 parent c08a124 commit c728e2d

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
@@ -6659,9 +6659,10 @@ def init(self, params: UseLockOrSemaphoreWorkflowParameters):
66596659
@workflow.run
66606660
async def run(
66616661
self,
6662-
params: UseLockOrSemaphoreWorkflowParameters,
6662+
params: Optional[UseLockOrSemaphoreWorkflowParameters],
66636663
) -> LockOrSemaphoreWorkflowConcurrencySummary:
66646664
# TODO: Use workflow init method when it exists.
6665+
assert params
66656666
self.init(params)
66666667
await asyncio.gather(
66676668
*(self.coroutine(f"{i}") for i in range(self.params.n_coroutines))
@@ -6716,6 +6717,7 @@ def __init__(self) -> None:
67166717
@workflow.run
67176718
async def run(
67186719
self,
6720+
_: Optional[UseLockOrSemaphoreWorkflowParameters] = None,
67196721
) -> LockOrSemaphoreWorkflowConcurrencySummary:
67206722
await workflow.wait_condition(lambda: self.workflow_may_exit)
67216723
return LockOrSemaphoreWorkflowConcurrencySummary(

0 commit comments

Comments
 (0)