Skip to content

Commit ca6c818

Browse files
author
Bryannah Hernandez
committed
pre-push fix
1 parent 1ec209c commit ca6c818

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -917,7 +917,7 @@ def _build_validations(self):
917917
raise ValueError(
918918
"IN_PROCESS mode is only supported for MMS/Transformers server in beta release."
919919
)
920-
920+
921921
if self.mode == Mode.IN_PROCESS and self.model_server == ModelServer.MMS:
922922
raise Exception(
923923
"IN_PROCESS mode is supported for MMS/Transformers server in beta release."

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ def test_validation_in_progress_mode_not_supported(self, mock_serveSettings):
8585
)
8686

8787
patch("sagemaker.serve.builder.model_builder._ServeSettings")
88+
8889
def test_validation_in_progress_mode_supported(self, mock_serveSettings):
8990
builder = ModelBuilder(model_server=ModelServer.MMS)
9091
self.assertRaisesRegex(

0 commit comments

Comments
 (0)