Skip to content

Commit df29ea3

Browse files
mufaddal-rohawalajerrypeng7773
authored andcommitted
fix: gpu integs CapacityError - fallback to available compute (aws#3008)
1 parent c9539bd commit df29ea3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/integ/test_tf.py

Lines changed: 5 additions & 0 deletions
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)