Skip to content

Commit 23a33a7

Browse files
authored
change: use p2 instead of p3 for the Horovod test (#1037)
1 parent 8b4140d commit 23a33a7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_horovod.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030

3131
@pytest.fixture(scope="module")
3232
def gpu_instance_type(request):
33-
return "ml.p3.2xlarge"
33+
return "ml.p2.xlarge"
3434

3535

3636
@pytest.mark.canary_quick
@@ -40,7 +40,7 @@ def test_hvd_cpu(sagemaker_session, cpu_instance_type, tmpdir):
4040

4141
@pytest.mark.canary_quick
4242
@pytest.mark.skipif(
43-
integ.test_region() in integ.HOSTING_NO_P3_REGIONS, reason="no ml.p3 instances in this region"
43+
integ.test_region() in integ.HOSTING_NO_P2_REGIONS, reason="no ml.p2 instances in this region"
4444
)
4545
def test_hvd_gpu(sagemaker_session, gpu_instance_type, tmpdir):
4646
__create_and_fit_estimator(sagemaker_session, gpu_instance_type, tmpdir)

0 commit comments

Comments
 (0)