From b41420977c13e385358bfdfa26aded066163f9c0 Mon Sep 17 00:00:00 2001 From: Rohan Narayan Date: Tue, 25 Mar 2025 18:49:04 +0000 Subject: [PATCH] chore: fix integ tests to use latest version of model --- tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py index 5e54c7551f..c9a39ac3dc 100644 --- a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py +++ b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py @@ -170,7 +170,7 @@ def test_jumpstart_gated_model(setup): model = JumpStartModel( model_id=model_id, - model_version="3.*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets + model_version="*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets role=get_sm_session().get_caller_identity_arn(), sagemaker_session=get_sm_session(), ) @@ -197,7 +197,7 @@ def test_jumpstart_gated_model_inference_component_enabled(setup): model = JumpStartModel( model_id=model_id, - model_version="3.*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets + model_version="*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets role=get_sm_session().get_caller_identity_arn(), sagemaker_session=get_sm_session(), )