File tree 1 file changed +8
-0
lines changed
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):
32
32
integ .test_region () not in integ .TRAINING_COMPILER_SUPPORTED_REGIONS ,
33
33
reason = "SageMaker Training Compiler is not supported in this region" ,
34
34
)
35
+ @pytest .mark .skipif (
36
+ integ .test_region () in integ .TRAINING_NO_P3_REGIONS ,
37
+ reason = "no ml.p3 instances in this region" ,
38
+ )
35
39
def test_huggingface_pytorch (
36
40
sagemaker_session ,
37
41
gpu_instance_type ,
@@ -78,6 +82,10 @@ def test_huggingface_pytorch(
78
82
integ .test_region () not in integ .TRAINING_COMPILER_SUPPORTED_REGIONS ,
79
83
reason = "SageMaker Training Compiler is not supported in this region" ,
80
84
)
85
+ @pytest .mark .skipif (
86
+ integ .test_region () in integ .TRAINING_NO_P3_REGIONS ,
87
+ reason = "no ml.p3 instances in this region" ,
88
+ )
81
89
def test_huggingface_tensorflow (
82
90
sagemaker_session ,
83
91
gpu_instance_type ,
You can’t perform that action at this time.
0 commit comments