minor fix

Bryannah Hernandez · Bryannah Hernandez · commit 826c5c405a0d · 2024-08-06T16:50:58.000-07:00
diff --git a/src/sagemaker/serve/builder/transformers_builder.py b/src/sagemaker/serve/builder/transformers_builder.py
@@ -239,11 +239,7 @@ def _transformers_model_builder_deploy_wrapper(self, *args, **kwargs) -> Type[Pr
             )
 
             self.modes[str(Mode.IN_PROCESS)].create_server(
-                self.image_uri,
-                timeout if timeout else DEFAULT_TIMEOUT,
-                None,
                 predictor,
-                self.pysdk_model.env,
             )
             return predictor
 

Original file line number	Diff line number	Diff line change
`@@ -239,11 +239,7 @@ def _transformers_model_builder_deploy_wrapper(self, args, *kwargs) -> Type[Pr`
`239`	`239`	`)`
`240`	`240`
`241`	`241`	`self.modes[str(Mode.IN_PROCESS)].create_server(`
`242`		`- self.image_uri,`
`243`		`- timeout if timeout else DEFAULT_TIMEOUT,`
`244`		`- None,`
`245`	`242`	`predictor,`
`246`		`- self.pysdk_model.env,`
`247`	`243`	`)`
`248`	`244`	`return predictor`
`249`	`245`