Add more descriptive error messages (triton-inference-server#13)

Author: Tabrizian

src/resources/startup.py

@@ -34,6 +34,7 @@
 import signal
 import time
 import struct
+import traceback
 
 from grpc_channelz.v1 import channelz
 from grpc_channelz.v1 import channelz_pb2
@@ -182,9 +183,10 @@ def Init(self, request, context):
             args = {x.key: x.value for x in request.args}
             try:
                 self.backend.initialize(args)
-            except tpb_utils.TritonModelException as e:
+            except Exception as e:
                 context.set_code(grpc.StatusCode.INTERNAL)
-                context.set_details(e.message())
+                tb = traceback.format_exc()
+                context.set_details(tb)
 
         return Empty()
 
@@ -195,9 +197,10 @@ def Fini(self, request, context):
         if hasattr(self.backend, 'finalize'):
             try:
                 self.backend.finalize()
-            except tpb_utils.TritonModelException as e:
+            except Exception as e:
                 context.set_code(grpc.StatusCode.INTERNAL)
-                context.set_details(e.message())
+                tb = traceback.format_exc()
+                context.set_details(tb)
 
         return Empty()
 
@@ -222,7 +225,7 @@ def Execute(self, request, context):
                 numpy_type = tpb_utils.triton_to_numpy_type(x.dtype)
 
                 # We need to deserialize TYPE_STRING
-                if numpy_type == np.object or numpy_type == np.bytes_:
+                if numpy_type == np.object_ or numpy_type == np.bytes_:
                     numpy_data = deserialize_bytes_tensor(x.raw_data)
                     tensor = tpb_utils.Tensor(x.name,
                                               numpy_data.reshape(x.dims))
@@ -250,7 +253,12 @@ def Execute(self, request, context):
             context.set_details('Backend does not implement `execute` method')
             return ExecuteResponse()
 
-        responses = self.backend.execute(inference_requests)
+        try:
+            responses = self.backend.execute(inference_requests)
+        except Exception as e:
+            context.set_code(grpc.StatusCode.INTERNAL)
+            tb = traceback.format_exc()
+            context.set_details(tb)
 
         # Make sure that number of InferenceResponse and InferenceRequest
         # objects match