Skip to content

Commit bbc0be4

Browse files
committed
Add type annotation
1 parent 783cd72 commit bbc0be4

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/nexus/test_workflow_caller_cancellation_types.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -94,9 +94,11 @@ async def cancel(
9494
# by the caller server. At that point, the caller server will write
9595
# NexusOperationCancelRequestCompleted. For TRY_CANCEL we want to prove that the nexus
9696
# op handle future can be resolved as cancelled before any of that.
97-
caller_wf = nexus.client().get_workflow_handle_for(
98-
CallerWorkflow.run,
99-
workflow_id=test_context.caller_workflow_id,
97+
caller_wf: WorkflowHandle[Any, CancellationResult] = (
98+
nexus.client().get_workflow_handle_for(
99+
CallerWorkflow.run,
100+
workflow_id=test_context.caller_workflow_id,
101+
)
100102
)
101103
await caller_wf.execute_update(
102104
CallerWorkflow.wait_caller_op_future_resolved

0 commit comments

Comments
 (0)