Skip to content

Commit 59245de

Browse files
fix: integs for training compiler in non-PDX regions
1 parent 412b633 commit 59245de

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/integ/test_training_compiler.py

+8
Original file line numberDiff line numberDiff 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+
)
3539
def 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+
)
8189
def test_huggingface_tensorflow(
8290
sagemaker_session,
8391
gpu_instance_type,

0 commit comments

Comments
 (0)