We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cfffb40 commit 9f813e2Copy full SHA for 9f813e2
tests/integ/sagemaker/lineage/test_artifact.py
@@ -78,10 +78,12 @@ def test_list(artifact_objs, sagemaker_session):
78
79
80
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)
+ # allow trial components to index, 30 seconds max
+ for i in range(3):
+ time.sleep(10)
+ trials = trial_associated_artifact.downstream_trials(sagemaker_session=sagemaker_session)
85
+ if len(trials) > 0:
86
+ break
87
88
assert len(trials) == 1
89
assert trial_obj.trial_name in trials
0 commit comments