File tree Expand file tree Collapse file tree 2 files changed +11
-2
lines changed
tests/unit/sagemaker/jumpstart Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Original file line number Diff line number Diff line change @@ -214,6 +214,10 @@ def _get_manifest_key_from_model_id_semantic_version(
214
214
raise KeyError (error_msg )
215
215
216
216
error_msg = f"Unable to find model manifest for '{ model_id } ' with version '{ version } '. "
217
+ error_msg += (
218
+ "Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/jumpstart.html"
219
+ " for updated list of models. "
220
+ )
217
221
218
222
other_model_id_version = self ._select_version (
219
223
"*" , versions_incompatible_with_sagemaker
Original file line number Diff line number Diff line change @@ -163,21 +163,26 @@ def test_jumpstart_cache_get_header():
163
163
)
164
164
assert (
165
165
"Unable to find model manifest for 'pytorch-ic-imagenet-inception-v3-classification-4' with "
166
- "version '3.*'. Consider using model ID 'pytorch-ic-imagenet-inception-v3-classification-4' with "
167
- "version '2.0.0'."
166
+ "version '3.*'. Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/jumpstart.html "
167
+ "for updated list of models. Consider using model ID 'pytorch-ic-imagenet-inception-v3-"
168
+ "classification-4' with version '2.0.0'."
168
169
) in str (e .value )
169
170
170
171
with pytest .raises (KeyError ) as e :
171
172
cache .get_header (model_id = "pytorch-ic-" , semantic_version_str = "*" )
172
173
assert (
173
174
"Unable to find model manifest for 'pytorch-ic-' with version '*'. "
175
+ "Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/jumpstart.html "
176
+ "for updated list of models. "
174
177
"Did you mean to use model ID 'pytorch-ic-imagenet-inception-v3-classification-4'?"
175
178
) in str (e .value )
176
179
177
180
with pytest .raises (KeyError ) as e :
178
181
cache .get_header (model_id = "tensorflow-ic-" , semantic_version_str = "*" )
179
182
assert (
180
183
"Unable to find model manifest for 'tensorflow-ic-' with version '*'. "
184
+ "Visit https://sagemaker.readthedocs.io/en/stable/doc_utils/jumpstart.html "
185
+ "for updated list of models. "
181
186
"Did you mean to use model ID 'tensorflow-ic-imagenet-inception-"
182
187
"v3-classification-4'?"
183
188
) in str (e .value )
You can’t perform that action at this time.
0 commit comments