test: retry context delete

Author: danabens

tests/integ/sagemaker/lineage/conftest.py

@@ -40,7 +40,7 @@
 from random import randint
 from botocore.exceptions import ClientError
 
-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
@@ -378,7 +378,11 @@ def model_obj(sagemaker_session):
 
     yield model
     time.sleep(SLEEP_TIME_SECONDS)
-    model.delete(disassociate=True)
+
+    def del_func():
+        return model.delete(disassociate=True)
+
+    retry(del_func, num_attempts=4)
 
 
 @pytest.fixture