Change env variable name

Author: nikhil-sk

src/sagemaker_inference/environment.py

@@ -80,7 +80,7 @@ def __init__(self):
         self._inference_http_port = os.environ.get(parameters.BIND_TO_PORT_ENV, DEFAULT_HTTP_PORT)
         self._management_http_port = os.environ.get(parameters.BIND_TO_PORT_ENV, DEFAULT_HTTP_PORT)
         self._safe_port_range = os.environ.get(parameters.SAFE_PORT_RANGE_ENV)
-        self._vmargs = os.environ.get(parameters.VMARGS, DEFAULT_VMARGS)
+        self._vmargs = os.environ.get(parameters.MODEL_SERVER_VMARGS, DEFAULT_VMARGS)
 
     @staticmethod
     def _parse_module_name(program_param):

src/sagemaker_inference/parameters.py

@@ -20,8 +20,8 @@
 DEFAULT_INVOCATIONS_ACCEPT_ENV = "SAGEMAKER_DEFAULT_INVOCATIONS_ACCEPT"  # type: str
 MODEL_SERVER_WORKERS_ENV = "SAGEMAKER_MODEL_SERVER_WORKERS"  # type: str
 MODEL_SERVER_TIMEOUT_ENV = "SAGEMAKER_MODEL_SERVER_TIMEOUT"  # type: str
+MODEL_SERVER_VMARGS = "SAGEMAKER_MODEL_SERVER_VMARGS"  # type: str
 STARTUP_TIMEOUT_ENV = "SAGEMAKER_STARTUP_TIMEOUT"  # type: str
 BIND_TO_PORT_ENV = "SAGEMAKER_BIND_TO_PORT"  # type: str
 SAFE_PORT_RANGE_ENV = "SAGEMAKER_SAFE_PORT_RANGE"  # type: str
 MULTI_MODEL_ENV = "SAGEMAKER_MULTI_MODEL"  # type: str
-VMARGS = "VMARGS"  # type: str

test/unit/test_environment.py

@@ -28,7 +28,7 @@
         parameters.DEFAULT_INVOCATIONS_ACCEPT_ENV: "text/html",
         parameters.BIND_TO_PORT_ENV: "1738",
         parameters.SAFE_PORT_RANGE_ENV: "1111-2222",
-        parameters.VMARGS: "-XX:-UseContainerSupport",
+        parameters.MODEL_SERVER_VMARGS: "-XX:-UseContainerSupport",
     },
     clear=True,
 )