diff --git a/src/sagemaker_inference/model_server.py b/src/sagemaker_inference/model_server.py index 47bd292..5808a5e 100644 --- a/src/sagemaker_inference/model_server.py +++ b/src/sagemaker_inference/model_server.py @@ -180,10 +180,15 @@ def _terminate(signo, frame): # pylint: disable=unused-argument def _install_requirements(): logger.info("installing packages from requirements.txt...") - pip_install_cmd = [sys.executable, "-m", "pip", "install", "-r", REQUIREMENTS_PATH] + if sys.version_info >= (3, 6): + pip_install_cmd = [sys.executable, "-m", "pip", "install", "-r", "requirements.txt"] + kw = {"cwd": code_dir} + else: + pip_install_cmd = [sys.executable, "-m", "pip", "install", "-r", REQUIREMENTS_PATH] + kw = {} try: - subprocess.check_call(pip_install_cmd) + subprocess.check_call(pip_install_cmd, **kw) except subprocess.CalledProcessError: logger.error("failed to install required packages, exiting") raise ValueError("failed to install required packages")