fix: skip gpu SageMaker test in regions with limited amount of p2/3 i… (aws#199)

* fix: skip gpu SageMaker test in regions with limited amount of p2/3 instances

Author: icywang86rui

test/conftest.py

@@ -20,6 +20,8 @@
 from sagemaker import LocalSession, Session
 from sagemaker.tensorflow import TensorFlow
 
+from test.integration import NO_P2_REGIONS, NO_P3_REGIONS
+
 logger = logging.getLogger(__name__)
 logging.getLogger('boto').setLevel(logging.INFO)
 logging.getLogger('botocore').setLevel(logging.INFO)
@@ -103,6 +105,13 @@ def skip_by_device_type(request, processor):
         pytest.skip('Skipping because running on \'{}\' instance'.format(processor))
 
 
+@pytest.fixture(autouse=True)
+def skip_gpu_instance_restricted_regions(region, instance_type):
+    if((region in NO_P2_REGIONS and instance_type.startswith('ml.p2')) or
+       (region in NO_P3_REGIONS and instance_type.startswith('ml.p3'))):
+        pytest.skip('Skipping GPU test in region {}'.format(region))
+
+
 @pytest.fixture(scope='session')
 def docker_image(docker_base_name, tag):
     return '{}:{}'.format(docker_base_name, tag)

test/integration/__init__.py

@@ -19,3 +19,8 @@
 logging.getLogger('botocore').setLevel(logging.INFO)
 
 RESOURCE_PATH = os.path.join(os.path.dirname(__file__), '..', 'resources')
+
+# these regions have some p2 and p3 instances, but not enough for automated testing
+NO_P2_REGIONS = ['ca-central-1', 'eu-central-1', 'eu-west-2', 'us-west-1']
+NO_P3_REGIONS = ['ap-southeast-1', 'ap-southeast-2', 'ap-south-1', 'ca-central-1',
+                 'eu-central-1', 'eu-west-2', 'us-west-1']