Skip to content

Commit 7f11ef4

Browse files
committed
fix sig
1 parent 2b74861 commit 7f11ef4

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

onnx_diagnostic/helpers/rt_helper.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,9 +115,11 @@ def make_feeds(
115115
# bool, int, float, onnxruntime does not support float, bool, int
116116
new_flat = []
117117
for i in flat:
118-
if isinstance(i, (bool, int)):
119-
i = np.array([i], dtype=np.int64)
118+
if isinstance(i, bool):
119+
i = np.array(i, dtype=np.bool_)
120+
elif isinstance(i, int):
121+
i = np.array(i, dtype=np.int64)
120122
elif isinstance(i, float):
121-
i = np.array([i], dtype=np.float32)
123+
i = np.array(i, dtype=np.float32)
122124
new_flat.append(i)
123125
return dict(zip(names, new_flat))

0 commit comments

Comments
 (0)