We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d180ab6 commit 6d1c24eCopy full SHA for 6d1c24e
src/sagemaker_mxnet_serving_container/handler_service.py
@@ -44,7 +44,10 @@ def __init__(self):
44
45
@staticmethod
46
def _user_module_transformer(model_dir=environment.model_dir):
47
- user_module = importlib.import_module(environment.Environment().module_name)
+ 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)
51
52
if hasattr(user_module, 'transform_fn'):
53
return Transformer(default_inference_handler=DefaultMXNetInferenceHandler())
0 commit comments