Fix to check p3

Author: benieric

tests/integ/__init__.py

@@ -68,7 +68,6 @@
     "me-south-1",
     "sa-east-1",
     "us-west-1",
-    "eu-west-1",  # not enough capacity
 ]
 TRAINING_NO_P3_REGIONS = [
     "af-south-1",
@@ -88,6 +87,7 @@
     "ap-south-1",
     "ap-northeast-2",  # it has p3, but not enough
     "us-east-2",  # it has p3, but not enough
+    "eu-west-1",  # it has p3, but not enough
 ]
 
 # EI is currently only supported in the following regions

tests/integ/test_huggingface.py

@@ -30,8 +30,8 @@
 
 @pytest.mark.release
 @pytest.mark.skipif(
-    tests.integ.test_region() in tests.integ.TRAINING_NO_P2_REGIONS,
-    reason="No P2 instances or low capacity in this region",
+    tests.integ.test_region() in tests.integ.TRAINING_NO_P3_REGIONS,
+    reason="No P3 instances or low capacity in this region",
 )
 def test_framework_processing_job_with_deps(
     sagemaker_session,
@@ -64,8 +64,8 @@ def test_framework_processing_job_with_deps(
 
 @pytest.mark.release
 @pytest.mark.skipif(
-    tests.integ.test_region() in tests.integ.TRAINING_NO_P2_REGIONS,
-    reason="No P2 instances or low capacity in this region",
+    tests.integ.test_region() in tests.integ.TRAINING_NO_P3_REGIONS,
+    reason="No P3 instances or low capacity in this region",
 )
 def test_huggingface_training(
     sagemaker_session,

tests/integ/test_training_compiler.py

@@ -91,8 +91,8 @@ def skip_if_incompatible(gpu_instance_type, request):
     ],
 )
 @pytest.mark.skipif(
-    integ.test_region() in integ.TRAINING_NO_P2_REGIONS,
-    reason="No P2 instances or low capacity in this region",
+    integ.test_region() in integ.TRAINING_NO_P3_REGIONS,
+    reason="No P3 instances or low capacity in this region",
 )
 def test_huggingface_pytorch(
     sagemaker_session,