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 <[email protected]>
aws · Dec 18, 2024 · cb45e72 · cb45e72
1 parent 31ea42c
commit cb45e72
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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: