Skip to content

Commit 3a8a2e7

Browse files
authored
fix: retry is_run assertion (#3793)
1 parent 9820b2d commit 3a8a2e7

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

tests/integ/sagemaker/experiments/test_run.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -134,11 +134,15 @@ def test_run_name_vs_trial_component_name_edge_cases(sagemaker_session, input_na
134134
) as run1:
135135
assert not run1._experiment.tags
136136
assert not run1._trial.tags
137-
is_run_tc = is_run_trial_component(
138-
trial_component_name=run1._trial_component.trial_component_name,
139-
sagemaker_session=sagemaker_session,
140-
)
141-
assert is_run_tc
137+
138+
def verify_is_run():
139+
is_run_tc = is_run_trial_component(
140+
trial_component_name=run1._trial_component.trial_component_name,
141+
sagemaker_session=sagemaker_session,
142+
)
143+
assert is_run_tc
144+
145+
retry_with_backoff(verify_is_run, 4)
142146

143147
with load_run(
144148
experiment_name=exp_name,

0 commit comments

Comments
 (0)