Skip to content

Commit 19c2a5f

Browse files
committed
change: cleanup code, remove redundant tests
1 parent 9eb82a4 commit 19c2a5f

File tree

15 files changed

+3
-695
lines changed

15 files changed

+3
-695
lines changed

src/sagemaker/jumpstart/artifacts.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -110,19 +110,19 @@ def _retrieve_image_uri(
110110
if framework is not None and framework != ecr_specs.framework:
111111
raise ValueError(
112112
f"Incorrect container framework '{framework}' for JumpStart model ID '{model_id}' "
113-
f"and version {model_version}'."
113+
f"and version '{model_version}'."
114114
)
115115

116116
if version is not None and version != ecr_specs.framework_version:
117117
raise ValueError(
118118
f"Incorrect container framework version '{version}' for JumpStart model ID "
119-
f"'{model_id}' and version {model_version}'."
119+
f"'{model_id}' and version '{model_version}'."
120120
)
121121

122122
if py_version is not None and py_version != ecr_specs.py_version:
123123
raise ValueError(
124124
f"Incorrect python version '{py_version}' for JumpStart model ID '{model_id}' "
125-
f"and version {model_version}'."
125+
f"and version '{model_version}'."
126126
)
127127

128128
base_framework_version_override: Optional[str] = None

tests/unit/sagemaker/model_uris/jumpstart/test_catboost.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

tests/unit/sagemaker/model_uris/jumpstart/test_huggingface.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

tests/unit/sagemaker/model_uris/jumpstart/test_lightgbm.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

tests/unit/sagemaker/model_uris/jumpstart/test_mxnet.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

tests/unit/sagemaker/model_uris/jumpstart/test_pytorch.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

tests/unit/sagemaker/model_uris/jumpstart/test_sklearn.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

tests/unit/sagemaker/model_uris/jumpstart/test_tensorflow.py

Lines changed: 0 additions & 49 deletions
This file was deleted.

0 commit comments

Comments
 (0)