Skip to content

Commit 9f1a10e

Browse files
committed
fix: write the mms config file to a subdir os base_dir instead of /etc
fix #72
1 parent d6d75cd commit 9f1a10e

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/sagemaker_inference/environment.py

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
model_dir = os.path.join(base_dir, "model") # type: str
3636
# str: the directory where models should be saved, e.g., /opt/ml/model/
3737

38+
config_dir = os.path.join(base_dir, "etc") # type: str
3839
code_dir = os.path.join(model_dir, "code") # type: str
3940
"""str: the path of the user's code directory, e.g., /opt/ml/model/code/"""
4041

src/sagemaker_inference/model_server.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,11 @@
2525

2626
import sagemaker_inference
2727
from sagemaker_inference import default_handler_service, environment, logging, utils
28-
from sagemaker_inference.environment import code_dir
28+
from sagemaker_inference.environment import code_dir, config_dir
2929

3030
logger = logging.get_logger()
3131

32-
MMS_CONFIG_FILE = os.path.join("/etc", "sagemaker-mms.properties")
32+
MMS_CONFIG_FILE = os.path.join(config_dir, "sagemaker-mms.properties")
3333
DEFAULT_HANDLER_SERVICE = default_handler_service.__name__
3434
DEFAULT_MMS_CONFIG_FILE = pkg_resources.resource_filename(
3535
sagemaker_inference.__name__, "/etc/default-mms.properties"

0 commit comments

Comments
 (0)