File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed
tests/unit/sagemaker/serve/builder Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change 67
67
class TestModelBuilder (unittest .TestCase ):
68
68
@patch ("sagemaker.serve.builder.model_builder._ServeSettings" )
69
69
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
+ )
71
78
self .assertRaisesRegex (
72
79
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." ,
74
82
builder .build ,
75
83
Mode .IN_PROCESS ,
76
84
mock_role_arn ,
You can’t perform that action at this time.
0 commit comments