Skip to content

Commit 8240261

Browse files
committed
fix: improve reliability of Run integration test
1 parent c02d74b commit 8240261

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/integ/sagemaker/experiments/test_run.py

Lines changed: 6 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,11 @@ 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 after
250+
# wait is complete resulting in run TC being updated, then when the above with statement
251+
# is exited another update trial component call is made _sometimes_ resulting in a ConflictException
252+
time.sleep(3)
253+
248254
_check_tc_status_when_exiting(
249255
trial_component_name=run._trial_component.trial_component_name,
250256
init_start_time=init_start_time,

0 commit comments

Comments
 (0)