From 0c555c18d2de958270c6fe12acdf25356c97f2a7 Mon Sep 17 00:00:00 2001 From: Dana Benson Date: Thu, 21 Oct 2021 09:48:09 -0700 Subject: [PATCH] test: retry context delete --- tests/integ/sagemaker/lineage/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integ/sagemaker/lineage/conftest.py b/tests/integ/sagemaker/lineage/conftest.py index a3b2b844ad..5e201eef42 100644 --- a/tests/integ/sagemaker/lineage/conftest.py +++ b/tests/integ/sagemaker/lineage/conftest.py @@ -43,7 +43,7 @@ ) from sagemaker.lineage.lineage_trial_component import LineageTrialComponent -from tests.integ.sagemaker.lineage.helpers import name, names +from tests.integ.sagemaker.lineage.helpers import name, names, retry SLEEP_TIME_SECONDS = 1 SLEEP_TIME_TWO_SECONDS = 2 @@ -400,7 +400,7 @@ def model_obj(sagemaker_session): yield model time.sleep(SLEEP_TIME_SECONDS) - model.delete(disassociate=True) + retry(lambda: model.delete(disassociate=True), num_attempts=4) @pytest.fixture