Skip to content

Commit eb6e8d5

Browse files
committed
change: skip p2 tests in ap-south-east
1 parent b6247a3 commit eb6e8d5

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

tests/integ/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
HOSTING_NO_P2_REGIONS = ['ca-central-1', 'eu-central-1', 'eu-west-2', 'us-west-1']
2929
HOSTING_NO_P3_REGIONS = ['ap-southeast-1', 'ap-southeast-2', 'ap-south-1', 'ca-central-1',
3030
'eu-central-1', 'eu-west-2', 'us-west-1']
31+
TRAINING_NO_P2_REGIONS = ['ap-southeast-1', 'ap-southeast-2']
3132

3233
# EI is currently only supported in the following regions
3334
# regions were derived from https://aws.amazon.com/machine-learning/elastic-inference/pricing/

tests/integ/test_tf_script_mode.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@
4242
'ml.c5.xlarge',
4343
pytest.param('ml.p2.xlarge',
4444
marks=pytest.mark.skipif(
45-
tests.integ.test_region() in tests.integ.HOSTING_NO_P2_REGIONS,
45+
tests.integ.test_region() in tests.integ.HOSTING_NO_P2_REGIONS
46+
or tests.integ.test_region() in tests.integ.TRAINING_NO_P2_REGIONS,
4647
reason='no ml.p2 instances in this region'))])
4748
def instance_type(request):
4849
return request.param

0 commit comments

Comments
 (0)