diff --git a/tests/integ/test_training_compiler.py b/tests/integ/test_training_compiler.py index b4f074745d..0d9653ea93 100644 --- a/tests/integ/test_training_compiler.py +++ b/tests/integ/test_training_compiler.py @@ -32,6 +32,10 @@ def gpu_instance_type(request): integ.test_region() not in integ.TRAINING_COMPILER_SUPPORTED_REGIONS, reason="SageMaker Training Compiler is not supported in this region", ) +@pytest.mark.skipif( + integ.test_region() in integ.TRAINING_NO_P3_REGIONS, + reason="no ml.p3 instances in this region", +) def test_huggingface_pytorch( sagemaker_session, gpu_instance_type, @@ -78,6 +82,10 @@ def test_huggingface_pytorch( integ.test_region() not in integ.TRAINING_COMPILER_SUPPORTED_REGIONS, reason="SageMaker Training Compiler is not supported in this region", ) +@pytest.mark.skipif( + integ.test_region() in integ.TRAINING_NO_P3_REGIONS, + reason="no ml.p3 instances in this region", +) def test_huggingface_tensorflow( sagemaker_session, gpu_instance_type,