File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,10 @@ def gpu_instance_type(request):
3232 integ .test_region () not in integ .TRAINING_COMPILER_SUPPORTED_REGIONS ,
3333 reason = "SageMaker Training Compiler is not supported in this region" ,
3434)
35+ @pytest .mark .skipif (
36+ integ .test_region () in integ .TRAINING_NO_P3_REGIONS ,
37+ reason = "no ml.p3 instances in this region" ,
38+ )
3539def test_huggingface_pytorch (
3640 sagemaker_session ,
3741 gpu_instance_type ,
@@ -78,6 +82,10 @@ def test_huggingface_pytorch(
7882 integ .test_region () not in integ .TRAINING_COMPILER_SUPPORTED_REGIONS ,
7983 reason = "SageMaker Training Compiler is not supported in this region" ,
8084)
85+ @pytest .mark .skipif (
86+ integ .test_region () in integ .TRAINING_NO_P3_REGIONS ,
87+ reason = "no ml.p3 instances in this region" ,
88+ )
8189def test_huggingface_tensorflow (
8290 sagemaker_session ,
8391 gpu_instance_type ,
You can’t perform that action at this time.
0 commit comments