Skip to content

Commit 1e24b0f

Browse files
committed
make default retry parameters static vars
1 parent c435184 commit 1e24b0f

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/sagemaker/workflow/retry.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@
1919

2020
from sagemaker.workflow.entities import Entity, DefaultEnumMeta, RequestType
2121

22+
23+
DEFAULT_BACKOFF_RATE = 2.0
24+
DEFAULT_INTERVAL_SECONDS = 1
2225
MAX_ATTEMPTS_CAP = 20
2326
MAX_EXPIRE_AFTER_MIN = 14400
2427

@@ -53,8 +56,8 @@ class RetryPolicy(Entity):
5356
to expire any further retry attempt (default: None)
5457
"""
5558

56-
backoff_rate: float = attr.ib(default=2.0)
57-
interval_seconds: int = attr.ib(default=1.0)
59+
backoff_rate: float = attr.ib(default=DEFAULT_BACKOFF_RATE)
60+
interval_seconds: int = attr.ib(default=DEFAULT_INTERVAL_SECONDS)
5861
max_attempts: int = attr.ib(default=None)
5962
expire_after_mins: int = attr.ib(default=None)
6063

0 commit comments

Comments
 (0)