fix: write the mms config file to a subdir os base_dir instead of /etc

fix #72
aws · Oct 30, 2020 · 9f1a10e · 9f1a10e
1 parent d6d75cd
commit 9f1a10e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/sagemaker_inference/environment.py b/src/sagemaker_inference/environment.py
@@ -35,6 +35,7 @@
     model_dir = os.path.join(base_dir, "model")  # type: str
 # str: the directory where models should be saved, e.g., /opt/ml/model/
 
+config_dir = os.path.join(base_dir, "etc")  # type: str
 code_dir = os.path.join(model_dir, "code")  # type: str
 """str: the path of the user's code directory, e.g., /opt/ml/model/code/"""
 

diff --git a/src/sagemaker_inference/model_server.py b/src/sagemaker_inference/model_server.py
@@ -25,11 +25,11 @@
 
 import sagemaker_inference
 from sagemaker_inference import default_handler_service, environment, logging, utils
-from sagemaker_inference.environment import code_dir
+from sagemaker_inference.environment import code_dir, config_dir
 
 logger = logging.get_logger()
 
-MMS_CONFIG_FILE = os.path.join("/etc", "sagemaker-mms.properties")
+MMS_CONFIG_FILE = os.path.join(config_dir, "sagemaker-mms.properties")
 DEFAULT_HANDLER_SERVICE = default_handler_service.__name__
 DEFAULT_MMS_CONFIG_FILE = pkg_resources.resource_filename(
     sagemaker_inference.__name__, "/etc/default-mms.properties"