Skip to content

Commit f88f07f

Browse files
committed
Update _create_retry_strategy to use kwargs
1 parent 59b9070 commit f88f07f

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

packages/smithy-core/src/smithy_core/retries.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,14 @@ async def resolve_retry_strategy(
4646
def _create_retry_strategy(
4747
self, retry_mode: RetryStrategyType, max_attempts: int | None
4848
) -> RetryStrategy:
49+
kwargs: dict[str, Any] = (
50+
{} if max_attempts is None else {"max_attempts": max_attempts}
51+
)
4952
match retry_mode:
5053
case "simple":
51-
if max_attempts is None:
52-
return SimpleRetryStrategy()
53-
else:
54-
return SimpleRetryStrategy(max_attempts=max_attempts)
54+
return SimpleRetryStrategy(**kwargs)
5555
case "standard":
56-
if max_attempts is None:
57-
return StandardRetryStrategy()
58-
else:
59-
return StandardRetryStrategy(max_attempts=max_attempts)
56+
return StandardRetryStrategy(**kwargs)
6057
case _:
6158
raise ValueError(f"Unknown retry mode: {retry_mode}")
6259

0 commit comments

Comments
 (0)