Skip to content

Commit afb374e

Browse files
vip-amznAstraea Quinn S
authored andcommitted
chore: fix step-with-retry test
1 parent b8921ac commit afb374e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/src/step_with_retry.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from aws_durable_execution_sdk_python.config import StepConfig
55
from aws_durable_execution_sdk_python.context import (
66
DurableContext,
7+
StepContext,
78
durable_step,
89
)
910
from aws_durable_execution_sdk_python.execution import durable_execution
@@ -14,7 +15,7 @@
1415

1516

1617
@durable_step
17-
def unreliable_operation() -> str:
18+
def unreliable_operation(_step_context: StepContext) -> str:
1819
failure_threshold = 0.5
1920
if random() > failure_threshold: # noqa: S311
2021
msg = "Random error occurred"

0 commit comments

Comments
 (0)