test: retry context delete

Author: danabens

tests/integ/sagemaker/lineage/conftest.py

@@ -28,7 +28,7 @@
 
 from smexperiments import trial_component, trial, experiment
 
-from tests.integ.sagemaker.lineage.helpers import name, names
+from tests.integ.sagemaker.lineage.helpers import name, names, retry
 
 SLEEP_TIME_SECONDS = 1
 
@@ -365,7 +365,7 @@ def model_obj(sagemaker_session):
 
     yield model
     time.sleep(SLEEP_TIME_SECONDS)
-    model.delete(disassociate=True)
+    retry(model.delete(disassociate=True), num_attempts=4)
 
 
 @pytest.fixture