Skip to content

Commit cb45e72

Browse files
malav-shastrimalavhs
andauthored
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 <malavhs@amazon.com>
1 parent 31ea42c commit cb45e72

File tree

1 file changed

+3
-0
lines changed
  • test/sagemaker_tests/huggingface/inference/integration/local

1 file changed

+3
-0
lines changed

test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from sagemaker.predictor import Predictor
2020
from sagemaker.serializers import JSONSerializer
2121
from sagemaker.deserializers import JSONDeserializer
22+
from packaging.version import Version
2223

2324
from ...integration import model_dir, ROLE, pt_model, tf_model
2425
from ...utils import local_mode_utils
@@ -60,6 +61,8 @@ def _assert_prediction(predictor):
6061
@pytest.mark.model("tiny-distilbert")
6162
@pytest.mark.team("sagemaker-1p-algorithms")
6263
def test_serve_json(docker_image, framework_version, sagemaker_local_session, instance_type):
64+
if "huggingface-pytorch" in docker_image and Version(framework_version) < Version("2.4"):
65+
pytest.skip("Skipping distilbert SM local tests for PT")
6366
with _predictor(
6467
model_dir, docker_image, framework_version, sagemaker_local_session, instance_type
6568
) as predictor:

0 commit comments

Comments
 (0)