From 94ec6ffece9abfdad28bb35e15453203756ea6bd Mon Sep 17 00:00:00 2001 From: pintaoz Date: Fri, 21 Mar 2025 15:22:19 -0700 Subject: [PATCH] Skip tests failed due to deprecated instance type --- tests/integ/sagemaker/serve/test_serve_model_builder_gpu.py | 4 ++++ tests/integ/sagemaker/serve/test_serve_transformers.py | 3 +++ 2 files changed, 7 insertions(+) diff --git a/tests/integ/sagemaker/serve/test_serve_model_builder_gpu.py b/tests/integ/sagemaker/serve/test_serve_model_builder_gpu.py index 8724fc5116..cf1eb65325 100644 --- a/tests/integ/sagemaker/serve/test_serve_model_builder_gpu.py +++ b/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 diff --git a/tests/integ/sagemaker/serve/test_serve_transformers.py b/tests/integ/sagemaker/serve/test_serve_transformers.py index 5f172f3edb..9405934474 100644 --- a/tests/integ/sagemaker/serve/test_serve_transformers.py +++ b/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