Skip to content

Commit b53de53

Browse files
author
Y Nguyen
committed
fix: fix loading user custom script
1 parent 0cc05b2 commit b53de53

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/sagemaker_inference/default_handler_service.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from __future__ import absolute_import
1515

1616
import os
17+
import sys
1718

1819
from sagemaker_inference.transformer import Transformer
1920

@@ -57,10 +58,7 @@ def initialize(self, context):
5758
model_dir = properties.get("model_dir")
5859

5960
# add model_dir/code to python path
60-
code_dir_path = "{}:".format(model_dir + "/code")
61-
if PYTHON_PATH_ENV in os.environ:
62-
os.environ[PYTHON_PATH_ENV] = code_dir_path + os.environ[PYTHON_PATH_ENV]
63-
else:
64-
os.environ[PYTHON_PATH_ENV] = code_dir_path
61+
code_dir_path = model_dir + "/code"
62+
sys.path.insert(1, code_dir_path)
6563

6664
self._service.validate_and_initialize(model_dir=model_dir, context=context)

0 commit comments

Comments
 (0)