@@ -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