Skip to content

Commit 0bd6aa8

Browse files
author
Jonathan Makunga
committed
Skip Alt Config integ tests as metadata aren't fully deployed.
1 parent 0e58562 commit 0bd6aa8

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed

requirements/extras/test_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,4 @@ accelerate>=0.24.1,<=0.27.0
3838
schema==0.7.5
3939
tensorflow>=2.1,<=2.16
4040
mlflow>=2.12.2,<2.13
41-
huggingface_hub>=0.23.4
41+
huggingface_hub>=0.23.4

tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,8 +354,12 @@ def test_register_gated_jumpstart_model(setup):
354354
assert response is not None
355355

356356

357+
@pytest.mark.skipif(
358+
True,
359+
reason="Only enable after metadata is fully deployed.",
360+
)
357361
def test_jumpstart_model_with_deployment_configs(setup):
358-
model_id = "meta-textgeneration-llama-2-7b-f"
362+
model_id = "meta-textgeneration-llama-2-13b"
359363

360364
model = JumpStartModel(
361365
model_id=model_id,

tests/integ/sagemaker/serve/test_serve_js_happy.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,8 @@ def test_happy_mms_sagemaker_endpoint(happy_mms_model_builder, gpu_instance_type
144144

145145

146146
@pytest.mark.skipif(
147-
PYTHON_VERSION_IS_NOT_310,
148-
reason="The goal of these test are to test the serving components of our feature",
147+
True,
148+
reason="Only enable after metadata is fully deployed.",
149149
)
150150
def test_js_model_with_deployment_configs(
151151
meta_textgeneration_llama_2_7b_f_schema,
@@ -157,7 +157,7 @@ def test_js_model_with_deployment_configs(
157157
role_arn = iam_client.get_role(RoleName=ROLE_NAME)["Role"]["Arn"]
158158

159159
model_builder = ModelBuilder(
160-
model="meta-textgeneration-llama-2-7b-f",
160+
model="meta-textgeneration-llama-2-13b",
161161
schema_builder=meta_textgeneration_llama_2_7b_f_schema,
162162
)
163163
configs = model_builder.list_deployment_configs()

0 commit comments

Comments
 (0)