Skip to content

Commit ca107d4

Browse files
authored
fix: improve reliability of Run integration test (#3745)
1 parent 4d1f901 commit ca107d4

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/integ/sagemaker/experiments/test_run.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import datetime
1616
import json
1717
import os
18+
import time
1819

1920
import pytest
2021

@@ -245,6 +246,12 @@ def test_run_from_local_and_train_job_and_all_exp_cfg_match(
245246
sagemaker_session=sagemaker_session,
246247
)
247248

249+
# the above estimator has wait=True but the job TC could still be receiving updates
250+
# after wait is complete resulting in run TC being updated, then when the above with
251+
# statement is exited another update trial component call is made _sometimes_
252+
# resulting in a ConflictException
253+
time.sleep(3)
254+
248255
_check_tc_status_when_exiting(
249256
trial_component_name=run._trial_component.trial_component_name,
250257
init_start_time=init_start_time,

0 commit comments

Comments
 (0)