Skip to content

Commit 1dc4a7d

Browse files
committed
[stress test] YT-27006: Fix retry policy
commit_hash:9620ab3dacf1107a914078ac3cfc59805ad34754
1 parent 2076505 commit 1dc4a7d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

yt/python/yt/wrapper/retries.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111

1212
def run_with_retries(
13-
action, retry_count=6, backoff=20.0, exceptions=(YtError,), except_action=None,
13+
action, retry_count=6, backoff=20.0, backoff_config=None, exceptions=(YtError,), except_action=None,
1414
backoff_action=None, additional_retriable_error_codes=None
1515
):
1616
if additional_retriable_error_codes is None:
@@ -22,7 +22,7 @@ def __init__(self):
2222
"enable": True,
2323
"count": retry_count,
2424
"total_timeout": None,
25-
"backoff": {"policy": "rounded_up_to_request_timeout"},
25+
"backoff": backoff_config or {"policy": "rounded_up_to_request_timeout"},
2626
"additional_retriable_error_codes": additional_retriable_error_codes,
2727
}
2828
super(SimpleRetrier, self).__init__(retry_config, backoff * 1000.0, exceptions)

0 commit comments

Comments
 (0)