Skip to content

Commit 6d1c24e

Browse files
authored
fix MME 'NoneType' object error
1 parent d180ab6 commit 6d1c24e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sagemaker_mxnet_serving_container/handler_service.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,10 @@ def __init__(self):
4444

4545
@staticmethod
4646
def _user_module_transformer(model_dir=environment.model_dir):
47-
user_module = importlib.import_module(environment.Environment().module_name)
47+
module_name = environment.Environment().module_name
48+
inference_script = model_dir + '/code' + '/{}.py'.format(module_name)
49+
spec = importlib.util.spec_from_file_location(module_name, inference_script)
50+
user_module = importlib.util.module_from_spec(spec)
4851

4952
if hasattr(user_module, 'transform_fn'):
5053
return Transformer(default_inference_handler=DefaultMXNetInferenceHandler())

0 commit comments

Comments
 (0)