Skip to content

Commit 40bf5d1

Browse files
committed
add back a few tests
1 parent 9a08f86 commit 40bf5d1

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/unit/sagemaker/jumpstart/test_utils.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1267,6 +1267,19 @@ def test_validate_model_id_and_get_type_invalid(
12671267
s3_client=mock_s3_client_value,
12681268
model_type=JumpStartModelType.PROPRIETARY,
12691269
)
1270+
1271+
self.assertIsNone(
1272+
utils.validate_model_id_and_get_type("dee", script=JumpStartScriptScope.TRAINING)
1273+
)
1274+
self.assertIsNone(
1275+
utils.validate_model_id_and_get_type("", script=JumpStartScriptScope.TRAINING)
1276+
)
1277+
self.assertIsNone(
1278+
utils.validate_model_id_and_get_type(None, script=JumpStartScriptScope.TRAINING)
1279+
)
1280+
self.assertIsNone(
1281+
utils.validate_model_id_and_get_type(set(), script=JumpStartScriptScope.TRAINING)
1282+
)
12701283

12711284
assert (
12721285
utils.validate_model_id_and_get_type("pytorch-eqa-bert-base-cased")

0 commit comments

Comments
 (0)