Skip to content

Commit 66c9f92

Browse files
committed
Log the error message in case module is not compatible with mxnet
1 parent 4f8ef93 commit 66c9f92

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/sagemaker_mxnet_serving_container/handler_service.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,6 @@ class HandlerService(DefaultHandlerService):
4141
Based on: https://github.com/awslabs/multi-model-server/blob/master/docs/custom_service.md
4242
4343
"""
44-
def __init__(self):
45-
# self._service = None
46-
transformer = Transformer(default_inference_handler=DefaultMXNetInferenceHandler())
47-
super(HandlerService, self).__init__(transformer=transformer)
48-
4944
@staticmethod
5045
def _user_module_transformer(model_dir=environment.model_dir):
5146
try:
@@ -83,5 +78,10 @@ def initialize(self, context):
8378
else:
8479
os.environ[PYTHON_PATH_ENV] = code_dir_path
8580

86-
self._service = self._user_module_transformer(model_dir)
81+
try:
82+
self._service = self._user_module_transformer(model_dir)
83+
except ValueError as e:
84+
logging.error(f"Error determining model. {str(e)}. For non-mxnet models, consider using pytorch-inference DLC that leverages TorchServe.")
85+
raise
86+
8787
super(HandlerService, self).initialize(context)

0 commit comments

Comments
 (0)