Skip to content

Commit 2237183

Browse files
fix: gpu integs CapacityError - fallback to available compute (#3008)
1 parent b3660b9 commit 2237183

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/integ/test_tf.py

+5
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,11 @@
4343

4444

4545
@pytest.mark.release
46+
@pytest.mark.skipif(
47+
tests.integ.test_region() in tests.integ.TRAINING_NO_P2_REGIONS
48+
and tests.integ.test_region() in tests.integ.TRAINING_NO_P3_REGIONS,
49+
reason="no ml.p2 or ml.p3 instances in this region",
50+
)
4651
def test_framework_processing_job_with_deps(
4752
sagemaker_session,
4853
gpu_instance_type_list,

0 commit comments

Comments
 (0)