Skip to content

Commit 47354f0

Browse files
authored
change: Add p2 instances to integ tests (#3852)
1 parent 217e8d1 commit 47354f0

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/integ/utils.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,15 @@
1919
from tests.conftest import NO_P3_REGIONS, NO_M4_REGIONS
2020
from sagemaker.exceptions import CapacityError
2121

22+
P2_INSTANCES = ["ml.p2.xlarge", "ml.p2.8xlarge", "ml.p2.16xlarge"]
23+
P3_INSTANCES = ["ml.p3.2xlarge"]
24+
2225

2326
def gpu_list(region):
2427
if region in NO_P3_REGIONS:
25-
return ["ml.p2.xlarge"]
28+
return P2_INSTANCES
2629
else:
27-
return ["ml.p3.2xlarge", "ml.p2.xlarge"]
30+
return [*P2_INSTANCES, *P3_INSTANCES]
2831

2932

3033
def cpu_list(region):

0 commit comments

Comments
 (0)