diff --git a/tests/integ/sagemaker/experiments/test_run.py b/tests/integ/sagemaker/experiments/test_run.py index cd0485a9f7..782f60aab7 100644 --- a/tests/integ/sagemaker/experiments/test_run.py +++ b/tests/integ/sagemaker/experiments/test_run.py @@ -15,6 +15,7 @@ import datetime import json import os +import time import pytest @@ -245,6 +246,12 @@ def test_run_from_local_and_train_job_and_all_exp_cfg_match( sagemaker_session=sagemaker_session, ) + # the above estimator has wait=True but the job TC could still be receiving updates + # after wait is complete resulting in run TC being updated, then when the above with + # statement is exited another update trial component call is made _sometimes_ + # resulting in a ConflictException + time.sleep(3) + _check_tc_status_when_exiting( trial_component_name=run._trial_component.trial_component_name, init_start_time=init_start_time,