Fix type boolean (triton-inference-server#35)

Tabrizian · web-flow · commit 7a7b869d9910 · 2021-02-10T20:03:58.000-08:00
diff --git a/src/python.cc b/src/python.cc
@@ -135,7 +135,6 @@ class ModelInstanceState : public BackendModelInstance {
  private:
   TRITONBACKEND_Model* triton_model_;
   pid_t interpreter_pid_;
-  std::vector<BackendMemory*> input_tensor_memories_;
 };
 
 class ModelState : public BackendModel {
@@ -333,12 +332,6 @@ ModelInstanceState::~ModelInstanceState()
     }
   }
 
-  // Remove input tensor memories
-  for (BackendMemory* mem : input_tensor_memories_) {
-    delete mem;
-  }
-  input_tensor_memories_.clear();
-
   stub.reset();
 
   int status;
diff --git a/src/resources/startup.py b/src/resources/startup.py
@@ -44,7 +44,8 @@
 import numpy as np
 
 from python_host_pb2 import *
-from python_host_pb2_grpc import PythonInterpreterServicer, add_PythonInterpreterServicer_to_server
+from python_host_pb2_grpc import \
+    PythonInterpreterServicer, add_PythonInterpreterServicer_to_server
 import grpc
 
 MAX_GRPC_MESSAGE_SIZE = 2147483647
@@ -72,9 +73,9 @@ def serialize_byte_tensor(input_tensor):
     if input_tensor.size == 0:
         return np.empty([0])
 
-    # If the input is a tensor of string/bytes objects, then must flatten those into
-    # a 1-dimensional array containing the 4-byte byte size followed by the
-    # actual element bytes. All elements are concatenated together in "C"
+    # If the input is a tensor of string/bytes objects, then must flatten those
+    # into a 1-dimensional array containing the 4-byte byte size followed by
+    # the actual element bytes. All elements are concatenated together in "C"
     # order.
     if (input_tensor.dtype == np.object) or (input_tensor.dtype.type
                                              == np.bytes_):
@@ -153,7 +154,9 @@ def __init__(self, module_path, *args, **kwargs):
         super(PythonInterpreterServicer, self).__init__(*args, **kwargs)
 
         module_path = Path(module_path).resolve()
-        # Add model parent directories so that relative and absolute import work
+
+        # Add model parent directories so that relative and absolute imports
+        # work.
         sys.path.append(str(module_path.parent))
         sys.path.append(str(module_path.parent.parent))
 
@@ -190,7 +193,7 @@ def Init(self, request, context):
             args = {x.key: x.value for x in request.args}
             try:
                 self.model_instance.initialize(args)
-            except Exception as e:
+            except Exception:
                 context.set_code(grpc.StatusCode.INTERNAL)
                 tb = traceback.format_exc()
                 context.set_details(tb)
diff --git a/src/resources/triton_python_backend_utils.py b/src/resources/triton_python_backend_utils.py
@@ -28,7 +28,7 @@
 
 TRITON_TO_NUMPY_TYPE = {
     # TRITONSERVER_TYPE_BOOL
-    1: np.bool,
+    1: np.bool_,
     # TRITONSERVER_TYPE_UINT8
     2: np.uint8,
     # TRITONSERVER_TYPE_UINT16
