@@ -74,11 +74,11 @@ def test_enabling_data_capture_on_endpoint_shows_correct_data_capture_status(
74
74
predictor .enable_data_capture ()
75
75
76
76
# 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
78
78
for _ in retries (
79
- max_retry_count = 50 ,
79
+ max_retry_count = 25 ,
80
80
exception_message_prefix = "Waiting for 'InService' endpoint status" ,
81
- seconds_to_sleep = 30 ,
81
+ seconds_to_sleep = 60 ,
82
82
):
83
83
new_endpoint = sagemaker_session .sagemaker_client .describe_endpoint (
84
84
EndpointName = predictor .endpoint
@@ -159,11 +159,11 @@ def test_disabling_data_capture_on_endpoint_shows_correct_data_capture_status(
159
159
predictor .disable_data_capture ()
160
160
161
161
# 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
163
163
for _ in retries (
164
- max_retry_count = 50 ,
164
+ max_retry_count = 25 ,
165
165
exception_message_prefix = "Waiting for 'InService' endpoint status" ,
166
- seconds_to_sleep = 30 ,
166
+ seconds_to_sleep = 60 ,
167
167
):
168
168
new_endpoint = sagemaker_session .sagemaker_client .describe_endpoint (
169
169
EndpointName = predictor .endpoint
@@ -228,11 +228,11 @@ def test_updating_data_capture_on_endpoint_shows_correct_data_capture_status(
228
228
)
229
229
230
230
# 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
232
232
for _ in retries (
233
- max_retry_count = 50 ,
233
+ max_retry_count = 25 ,
234
234
exception_message_prefix = "Waiting for 'InService' endpoint status" ,
235
- seconds_to_sleep = 30 ,
235
+ seconds_to_sleep = 60 ,
236
236
):
237
237
new_endpoint = sagemaker_session .sagemaker_client .describe_endpoint (
238
238
EndpointName = predictor .endpoint
0 commit comments