Skip to content

Commit 6542942

Browse files
Fix np.object for TYPE_STRING (triton-inference-server#31)
Co-authored-by: francescoclarifai <[email protected]>
1 parent 9e89c10 commit 6542942

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/resources/startup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ def Execute(self, request, context):
305305
output_shape = output_np_array.shape
306306

307307
# We need to serialize TYPE_STRING
308-
if output_np_array.dtype.type is np.object or output_np_array.dtype.type is np.bytes_:
308+
if output_np_array.dtype == np.object or output_np_array.dtype.type is np.bytes_:
309309
output_np_array = serialize_byte_tensor(output_np_array)
310310

311311
tensor = Tensor(name=output_tensor.name(),

0 commit comments

Comments
 (0)