Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 9 additions & 9 deletions tests/integ/test_data_capture_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,11 +74,11 @@ def test_enabling_data_capture_on_endpoint_shows_correct_data_capture_status(
predictor.enable_data_capture()

# Wait for endpoint to finish updating
# Endpoint update takes ~7min. 50 retries * 30s sleeps = 25min timeout
# Endpoint update takes ~7min. 25 retries * 60s sleeps = 25min timeout
for _ in retries(
max_retry_count=50,
max_retry_count=25,
exception_message_prefix="Waiting for 'InService' endpoint status",
seconds_to_sleep=30,
seconds_to_sleep=60,
):
new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
EndpointName=predictor.endpoint
Expand Down Expand Up @@ -159,11 +159,11 @@ def test_disabling_data_capture_on_endpoint_shows_correct_data_capture_status(
predictor.disable_data_capture()

# Wait for endpoint to finish updating
# Endpoint update takes ~7min. 50 retries * 30s sleeps = 25min timeout
# Endpoint update takes ~7min. 25 retries * 60s sleeps = 25min timeout
for _ in retries(
max_retry_count=50,
max_retry_count=25,
exception_message_prefix="Waiting for 'InService' endpoint status",
seconds_to_sleep=30,
seconds_to_sleep=60,
):
new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
EndpointName=predictor.endpoint
Expand Down Expand Up @@ -228,11 +228,11 @@ def test_updating_data_capture_on_endpoint_shows_correct_data_capture_status(
)

# Wait for endpoint to finish updating
# Endpoint update takes ~7min. 50 retries * 30s sleeps = 25min timeout
# Endpoint update takes ~7min. 25 retries * 60s sleeps = 25min timeout
for _ in retries(
max_retry_count=50,
max_retry_count=25,
exception_message_prefix="Waiting for 'InService' endpoint status",
seconds_to_sleep=30,
seconds_to_sleep=60,
):
new_endpoint = sagemaker_session.sagemaker_client.describe_endpoint(
EndpointName=predictor.endpoint
Expand Down