Skip to content

Commit 84a711b

Browse files
Captainiaakrishna1995
authored andcommitted
added one more test
1 parent 749e168 commit 84a711b

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
@@ -1291,6 +1291,11 @@ def test_validate_model_id_and_get_type_invalid(
12911291
model_type=JumpStartModelType.OPEN_WEIGHTS,
12921292
)
12931293

1294+
with pytest.raises(ValueError):
1295+
utils.validate_model_id_and_get_type(
1296+
"ai21-summarization", script=JumpStartScriptScope.TRAINING
1297+
)
1298+
12941299

12951300
class TestGetModelIdVersionFromResourceArn(TestCase):
12961301
def test_no_model_id_no_version_found(self):

0 commit comments

Comments
 (0)