File tree Expand file tree Collapse file tree 1 file changed +13
-0
lines changed
tests/unit/sagemaker/jumpstart Expand file tree Collapse file tree 1 file changed +13
-0
lines changed Original file line number Diff line number Diff line change @@ -1267,6 +1267,19 @@ def test_validate_model_id_and_get_type_invalid(
1267
1267
s3_client = mock_s3_client_value ,
1268
1268
model_type = JumpStartModelType .PROPRIETARY ,
1269
1269
)
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
+ )
1270
1283
1271
1284
assert (
1272
1285
utils .validate_model_id_and_get_type ("pytorch-eqa-bert-base-cased" )
You can’t perform that action at this time.
0 commit comments