File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
src/sagemaker_mxnet_serving_container Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -41,11 +41,6 @@ class HandlerService(DefaultHandlerService):
41
41
Based on: https://github.com/awslabs/multi-model-server/blob/master/docs/custom_service.md
42
42
43
43
"""
44
- def __init__ (self ):
45
- # self._service = None
46
- transformer = Transformer (default_inference_handler = DefaultMXNetInferenceHandler ())
47
- super (HandlerService , self ).__init__ (transformer = transformer )
48
-
49
44
@staticmethod
50
45
def _user_module_transformer (model_dir = environment .model_dir ):
51
46
try :
@@ -83,5 +78,10 @@ def initialize(self, context):
83
78
else :
84
79
os .environ [PYTHON_PATH_ENV ] = code_dir_path
85
80
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
+
87
87
super (HandlerService , self ).initialize (context )
You can’t perform that action at this time.
0 commit comments