Skip to content

Commit 1121f47

Browse files
author
Bryannah Hernandez
committed
unit test, test mb
1 parent 495c7b4 commit 1121f47

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/unit/sagemaker/serve/builder/test_model_builder.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,10 +67,18 @@
6767
class TestModelBuilder(unittest.TestCase):
6868
@patch("sagemaker.serve.builder.model_builder._ServeSettings")
6969
def test_validation_in_progress_mode_not_supported(self, mock_serveSettings):
70-
builder = ModelBuilder()
70+
builder = ModelBuilder(
71+
ModelServer.TORCHSERVE,
72+
ModelServer.TRITON,
73+
ModelServer.DJL_SERVING,
74+
ModelServer.TENSORFLOW_SERVING,
75+
ModelServer.TGI,
76+
ModelServer.TEI,
77+
)
7178
self.assertRaisesRegex(
7279
Exception,
73-
"IN_PROCESS mode is not supported yet!",
80+
"IN_PROCESS mode is not supported yet for model server. It is "
81+
"supported for MMS/Transformers server in beta release.",
7482
builder.build,
7583
Mode.IN_PROCESS,
7684
mock_role_arn,

0 commit comments

Comments
 (0)