Skip to content

Commit 9f813e2

Browse files
authored
fix: increase time allowed for trial components to index (#2090)
1 parent cfffb40 commit 9f813e2

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/integ/sagemaker/lineage/test_artifact.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,10 +78,12 @@ def test_list(artifact_objs, sagemaker_session):
7878

7979

8080
def test_downstream_trials(trial_associated_artifact, trial_obj, sagemaker_session):
81-
# wait for TC to index
82-
time.sleep(3)
83-
84-
trials = trial_associated_artifact.downstream_trials(sagemaker_session=sagemaker_session)
81+
# allow trial components to index, 30 seconds max
82+
for i in range(3):
83+
time.sleep(10)
84+
trials = trial_associated_artifact.downstream_trials(sagemaker_session=sagemaker_session)
85+
if len(trials) > 0:
86+
break
8587

8688
assert len(trials) == 1
8789
assert trial_obj.trial_name in trials

0 commit comments

Comments
 (0)