Skip to content

Commit 8419175

Browse files
author
Bryannah Hernandez
committed
minor fix
1 parent 2d5ce13 commit 8419175

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/sagemaker/serve/builder/requirements_manager.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,10 @@ def detect_file_exists(self, dependencies: str = None) -> str:
5151
def _install_requirements_txt(self):
5252
"""Install requirements.txt file using pip"""
5353
logger.info("Running command to pip install")
54-
subprocess.run("pip install -r /home/ec2-user/SageMaker/require.txt")
54+
subprocess.run(
55+
"pip install -r require.txt",
56+
shell=True
57+
)
5558
logger.info("Command ran successfully")
5659

5760
def _update_conda_env_in_path(self):

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)