Skip to content

Commit 511df55

Browse files
author
Bryannah Hernandez
committed
putting 6 args
1 parent de054b5 commit 511df55

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/sagemaker/serve/mode/in_process_mode.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,11 @@ def prepare(self):
6363

6464
def create_server(
6565
self,
66+
image: str,
67+
secret_key: str,
6668
predictor: PredictorBase,
69+
env_vars: Dict[str, str] = None,
70+
model_path: str = None,
6771
):
6872
"""Creating the server and checking ping health."""
6973
logger.info("Waiting for model server %s to start up...", self.model_server)

0 commit comments

Comments
 (0)