From 9ddd474e6f1d0e2a35c2f2212342a8c7cbbd8916 Mon Sep 17 00:00:00 2001 From: Mohamed Zeidan Date: Wed, 25 Mar 2026 12:37:23 -0700 Subject: [PATCH] fixed integ tests --- .../integ/sagemaker/jumpstart/model/test_jumpstart_model.py | 5 ++--- .../private_hub/model/test_jumpstart_private_hub_model.py | 2 +- tests/integ/sagemaker/serve/test_schema_builder.py | 1 - 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py index c9a39ac3dc..c8b89db7b6 100644 --- a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py +++ b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py @@ -87,7 +87,7 @@ def test_non_prepacked_jumpstart_model(setup): def test_prepacked_jumpstart_model(setup): - model_id = "huggingface-txt2img-conflictx-complex-lineart" + model_id = "model-txt2img-stabilityai-stable-diffusion-v2-1-base" model = JumpStartModel( model_id=model_id, @@ -251,11 +251,10 @@ def test_instantiating_model(mock_warning_logger, setup): def test_jumpstart_model_register(setup): - model_id = "huggingface-txt2img-conflictx-complex-lineart" + model_id = "model-txt2img-stabilityai-stable-diffusion-v2-1-base" model = JumpStartModel( model_id=model_id, - model_version="1.1.0", role=get_sm_session().get_caller_identity_arn(), sagemaker_session=get_sm_session(), ) diff --git a/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py b/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py index c7e039693b..76334330f5 100644 --- a/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py +++ b/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py @@ -39,7 +39,7 @@ TEST_MODEL_IDS = { "catboost-classification-model", - "huggingface-txt2img-conflictx-complex-lineart", + "model-txt2img-stabilityai-stable-diffusion-v2-1-base", "meta-textgeneration-llama-2-7b", "meta-textgeneration-llama-3-2-1b", "catboost-regression-model", diff --git a/tests/integ/sagemaker/serve/test_schema_builder.py b/tests/integ/sagemaker/serve/test_schema_builder.py index 6d3e8281d5..1a50678cd7 100644 --- a/tests/integ/sagemaker/serve/test_schema_builder.py +++ b/tests/integ/sagemaker/serve/test_schema_builder.py @@ -82,7 +82,6 @@ def test_model_builder_negative_path(sagemaker_session): None, ), ("HuggingFaceH4/zephyr-7b-beta", "text-generation", "ml.g5.2xlarge", 900), - ("HuggingFaceH4/zephyr-7b-alpha", "text-generation", "ml.g5.2xlarge", 900), ], ) def test_model_builder_happy_path_with_task_provided_local_schema_mode(