Skip to content

Commit 65995d5

Browse files
committed
Fix mypy-specific type check error
1 parent bbb6964 commit 65995d5

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
@@ -5909,9 +5909,10 @@ def init(self, params: UseLockOrSemaphoreWorkflowParameters):
59095909
@workflow.run
59105910
async def run(
59115911
self,
5912-
params: UseLockOrSemaphoreWorkflowParameters,
5912+
params: Optional[UseLockOrSemaphoreWorkflowParameters],
59135913
) -> LockOrSemaphoreWorkflowConcurrencySummary:
59145914
# TODO: Use workflow init method when it exists.
5915+
assert params
59155916
self.init(params)
59165917
await asyncio.gather(
59175918
*(self.coroutine(f"{i}") for i in range(self.params.n_coroutines))
@@ -5966,6 +5967,7 @@ def __init__(self) -> None:
59665967
@workflow.run
59675968
async def run(
59685969
self,
5970+
_: Optional[UseLockOrSemaphoreWorkflowParameters] = None,
59695971
) -> LockOrSemaphoreWorkflowConcurrencySummary:
59705972
await workflow.wait_condition(lambda: self.workflow_may_exit)
59715973
return LockOrSemaphoreWorkflowConcurrencySummary(

0 commit comments

Comments
 (0)