Skip to content

Commit 2affabb

Browse files
author
Bryannah Hernandez
committed
changes
1 parent 8419175 commit 2affabb

File tree

3 files changed

+3
-13
lines changed

3 files changed

+3
-13
lines changed

src/sagemaker/serve/builder/requirements_manager.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -51,19 +51,13 @@ 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(
55-
"pip install -r require.txt",
56-
shell=True
57-
)
54+
subprocess.run("pip install -r require.txt", shell=True)
5855
logger.info("Command ran successfully")
5956

6057
def _update_conda_env_in_path(self):
6158
"""Update conda env using conda yml file"""
6259
print("Updating conda env")
63-
subprocess.run(
64-
"conda env update -f conda_in_process.yml",
65-
shell=True
66-
)
60+
subprocess.run("conda env update -f conda_in_process.yml", shell=True)
6761
print("Conda env updated successfully")
6862

6963
def _get_active_conda_env_name(self) -> str:

src/sagemaker/serve/builder/transformers_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
from abc import ABC, abstractmethod
1818
from typing import Type
1919
from pathlib import Path
20-
from packaging.version import Version
2120
import subprocess
21+
from packaging.version import Version
2222

2323
from sagemaker.model import Model
2424
from sagemaker import image_uris

src/sagemaker/serve/mode/in_process_mode.py

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

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

0 commit comments

Comments
 (0)