skip p2 instance tests in eu-west-1

Author: benieric

tests/integ/__init__.py

@@ -68,6 +68,7 @@
     "me-south-1",
     "sa-east-1",
     "us-west-1",
+    "eu-west-1",  # not enough capacity
 ]
 TRAINING_NO_P3_REGIONS = [
     "af-south-1",

tests/integ/test_huggingface.py

@@ -29,7 +29,10 @@
 
 
 @pytest.mark.release
-@pytest.mark.flaky(reruns=3, reruns_delay=5)
+@pytest.mark.skipif(
+    tests.integ.test_region() in tests.integ.TRAINING_NO_P2_REGIONS,
+    reason="No P2 instances or low capacity in this region",
+)
 def test_framework_processing_job_with_deps(
     sagemaker_session,
     huggingface_training_latest_version,
@@ -60,7 +63,10 @@ def test_framework_processing_job_with_deps(
 
 
 @pytest.mark.release
-@pytest.mark.flaky(reruns=3, reruns_delay=5)
+@pytest.mark.skipif(
+    tests.integ.test_region() in tests.integ.TRAINING_NO_P2_REGIONS,
+    reason="No P2 instances or low capacity in this region",
+)
 def test_huggingface_training(
     sagemaker_session,
     huggingface_training_latest_version,

tests/integ/test_training_compiler.py

@@ -90,7 +90,10 @@ def skip_if_incompatible(gpu_instance_type, request):
         pytest.param("ml.p3.16xlarge", 2),
     ],
 )
-@pytest.mark.flaky(rerun=3, rerun_delay=5)
+@pytest.mark.skipif(
+    integ.test_region() in integ.TRAINING_NO_P2_REGIONS,
+    reason="No P2 instances or low capacity in this region",
+)
 def test_huggingface_pytorch(
     sagemaker_session,
     gpu_instance_type,