Skip to content

Commit 93dc590

Browse files
authored
change: reduce test's describe frequency to eliminate throttling error. (#1197)
1 parent 95c396d commit 93dc590

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/integ/test_data_capture_config.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -74,11 +74,11 @@ def test_enabling_data_capture_on_endpoint_shows_correct_data_capture_status(
7474
predictor.enable_data_capture()
7575

7676
# Wait for endpoint to finish updating
77-
# Endpoint update takes ~7min. 50 retries * 30s sleeps = 25min timeout
77+
# Endpoint update takes ~7min. 25 retries * 60s sleeps = 25min timeout
7878
for _ in retries(
79-
max_retry_count=50,
79+
max_retry_count=25,
8080
exception_message_prefix="Waiting for 'InService' endpoint status",
81-
seconds_to_sleep=30,
81+
seconds_to_sleep=60,
8282
):
8383
new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
8484
EndpointName=predictor.endpoint
@@ -159,11 +159,11 @@ def test_disabling_data_capture_on_endpoint_shows_correct_data_capture_status(
159159
predictor.disable_data_capture()
160160

161161
# Wait for endpoint to finish updating
162-
# Endpoint update takes ~7min. 50 retries * 30s sleeps = 25min timeout
162+
# Endpoint update takes ~7min. 25 retries * 60s sleeps = 25min timeout
163163
for _ in retries(
164-
max_retry_count=50,
164+
max_retry_count=25,
165165
exception_message_prefix="Waiting for 'InService' endpoint status",
166-
seconds_to_sleep=30,
166+
seconds_to_sleep=60,
167167
):
168168
new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
169169
EndpointName=predictor.endpoint
@@ -228,11 +228,11 @@ def test_updating_data_capture_on_endpoint_shows_correct_data_capture_status(
228228
)
229229

230230
# Wait for endpoint to finish updating
231-
# Endpoint update takes ~7min. 50 retries * 30s sleeps = 25min timeout
231+
# Endpoint update takes ~7min. 25 retries * 60s sleeps = 25min timeout
232232
for _ in retries(
233-
max_retry_count=50,
233+
max_retry_count=25,
234234
exception_message_prefix="Waiting for 'InService' endpoint status",
235-
seconds_to_sleep=30,
235+
seconds_to_sleep=60,
236236
):
237237
new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
238238
EndpointName=predictor.endpoint

0 commit comments

Comments
 (0)