Skip to content

Commit abd4c82

Browse files
Narrohagmollyheamazon
authored andcommitted
chore: fix integ tests to use latest version of model (#5104)
1 parent 1191050 commit abd4c82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def test_jumpstart_gated_model(setup):
170170

171171
model = JumpStartModel(
172172
model_id=model_id,
173-
model_version="3.*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets
173+
model_version="*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets
174174
role=get_sm_session().get_caller_identity_arn(),
175175
sagemaker_session=get_sm_session(),
176176
)
@@ -197,7 +197,7 @@ def test_jumpstart_gated_model_inference_component_enabled(setup):
197197

198198
model = JumpStartModel(
199199
model_id=model_id,
200-
model_version="3.*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets
200+
model_version="*", # version >=3.0.0 stores artifacts in jumpstart-private-cache-* buckets
201201
role=get_sm_session().get_caller_identity_arn(),
202202
sagemaker_session=get_sm_session(),
203203
)

0 commit comments

Comments
 (0)