Skip tests failed due to deprecated instance type (aws#5097)

Co-authored-by: pintaoz <[email protected]>

Author: 2 people

tests/integ/sagemaker/serve/test_serve_model_builder_gpu.py

@@ -96,6 +96,8 @@ def model_builder(request):
 def test_non_text_generation_model_single_GPU(
     sagemaker_session, model_builder, model_input, **kwargs
 ):
+    if kwargs["instance_type"] == "ml.p2.xlarge":
+        pytest.skip("Instance type ml.p2.xlarge has been deprecated")
     iam_client = sagemaker_session.boto_session.client("iam")
     role_arn = iam_client.get_role(RoleName="SageMakerRole")["Role"]["Arn"]
     model = model_builder.build(role_arn=role_arn, sagemaker_session=sagemaker_session)
@@ -147,6 +149,8 @@ def test_non_text_generation_model_single_GPU(
 def test_non_text_generation_model_multi_GPU(
     sagemaker_session, model_builder, model_input, **kwargs
 ):
+    if kwargs["instance_type"] == "ml.p2.xlarge":
+        pytest.skip("Instance type ml.p2.xlarge has been deprecated")
     iam_client = sagemaker_session.boto_session.client("iam")
     role_arn = iam_client.get_role(RoleName="SageMakerRole")["Role"]["Arn"]
     caught_ex = None

tests/integ/sagemaker/serve/test_serve_transformers.py

@@ -97,6 +97,9 @@ def model_builder(request):
 def test_pytorch_transformers_sagemaker_endpoint(
     sagemaker_session, model_builder, model_input, **kwargs
 ):
+    if kwargs["instance_type"] == "ml.p2.xlarge":
+        pytest.skip("Instance type ml.p2.xlarge has been deprecated")
+
     logger.info("Running in SAGEMAKER_ENDPOINT mode...")
     caught_ex = None