Skip to content

Commit 9c29a17

Browse files
committed
add test
1 parent 84a711b commit 9c29a17

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/unit/sagemaker/jumpstart/test_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1215,6 +1215,11 @@ def test_validate_model_id_and_get_type_open_weights(
12151215
]
12161216

12171217
mock_get_model_specs.return_value = Mock(training_supported=True)
1218+
self.assertIsNone(
1219+
utils.validate_model_id_and_get_type(
1220+
"invalid", script=JumpStartScriptScope.TRAINING
1221+
)
1222+
)
12181223
assert (
12191224
utils.validate_model_id_and_get_type("bee", script=JumpStartScriptScope.TRAINING)
12201225
== JumpStartModelType.OPEN_WEIGHTS

0 commit comments

Comments
 (0)