test: retry context delete

Author: danabens

tests/integ/sagemaker/lineage/conftest.py

@@ -33,7 +33,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
@@ -371,7 +371,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