Add data type for python script

onnx · Sep 6, 2023 · 36f8bca · 36f8bca
1 parent 748be0c
commit 36f8bca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/backend/inference_backend.py b/test/backend/inference_backend.py
@@ -1375,6 +1375,7 @@ def JniExecutionSession(jar_name, inputs):
         "f2": np.float16,
         "f4": np.float32,
         "f8": np.float64,
+        "U25": np.str_,
     }
 
     # print('stdout=' + str(procStdout), file=sys.stderr)