Merge pull request #1404 from shrinath-suresh/ts-kservev2-bytes-fix

Kserve V2 - Fix for Bytes output

Author: lxning

ts/torch_handler/request_envelope/kservev2.py

@@ -30,6 +30,10 @@
 # NOTE: numpy has more types than v2 protocol
 _NumpyToDatatype["object"] = "BYTES"
 
+# Adding support for unicode string
+# Ref: https://numpy.org/doc/stable/reference/arrays.dtypes.html
+_NumpyToDatatype["U"] = "BYTES"
+
 
 def _to_dtype(datatype: str) -> "np.dtype":
     dtype = _DatatypeToNumpy[datatype]
@@ -38,6 +42,8 @@ def _to_dtype(datatype: str) -> "np.dtype":
 
 def _to_datatype(dtype: np.dtype) -> str:
     as_str = str(dtype)
+    if as_str not in _NumpyToDatatype:
+        as_str = getattr(dtype, "kind")
     datatype = _NumpyToDatatype[as_str]
 
     return datatype