From cb45e7218d56902d932625ef3198aec50f6273c8 Mon Sep 17 00:00:00 2001 From: Malav Shastri <57682969+malav-shastri@users.noreply.github.com> Date: Wed, 18 Dec 2024 09:46:22 -0500 Subject: [PATCH] skip hf_pt-inference-latest-dlc-sagemaker-local-test temporarily for HF PT Inference latest image (#4465) * skip hf_pt-inference-latest-dlc-sagemaker-local-test temporarily * revert dlc_developer_config.toml --------- Co-authored-by: malavhs --- .../huggingface/inference/integration/local/test_serving.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py b/test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py index 85c7cdc51b50..af4d2c8648a1 100644 --- a/test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py +++ b/test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py @@ -19,6 +19,7 @@ from sagemaker.predictor import Predictor from sagemaker.serializers import JSONSerializer from sagemaker.deserializers import JSONDeserializer +from packaging.version import Version from ...integration import model_dir, ROLE, pt_model, tf_model from ...utils import local_mode_utils @@ -60,6 +61,8 @@ def _assert_prediction(predictor): @pytest.mark.model("tiny-distilbert") @pytest.mark.team("sagemaker-1p-algorithms") def test_serve_json(docker_image, framework_version, sagemaker_local_session, instance_type): + if "huggingface-pytorch" in docker_image and Version(framework_version) < Version("2.4"): + pytest.skip("Skipping distilbert SM local tests for PT") with _predictor( model_dir, docker_image, framework_version, sagemaker_local_session, instance_type ) as predictor: