Skip to content

Commit 5620eb8

Browse files
committed
fix: skip p2 test of tf script mode in regions without p2 instances
1 parent 3c77951 commit 5620eb8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integ/test_tf_script_mode.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@ def instance_type(request):
4040
return request.param
4141

4242

43+
@pytest.mark.skipif(integ.test_region() in integ.HOSTING_NO_P2_REGIONS,
44+
reason='no ml.p2 instances in these regions')
4345
@pytest.mark.skipif(integ.PYTHON_VERSION != 'py3', reason="Script Mode tests are only configured to run with Python 3")
4446
def test_mnist(sagemaker_session, instance_type):
4547
estimator = TensorFlow(entry_point=SCRIPT,

0 commit comments

Comments
 (0)