Skip to content

Commit ced1fe9

Browse files
committedAug 30, 2024
fix
1 parent 202562c commit ced1fe9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed
 

‎onnx_array_api/ort/ort_profile.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,9 @@ def ort_profile(
101101
if providers is None:
102102
providers = ["CPUExecutionProvider"]
103103
sess = InferenceSession(obj, sess_options, providers=providers, **kwargs)
104-
first = next(feeds.values())
104+
for v in feeds.values():
105+
first = v
106+
break
105107

106108
if isinstance(first, numpy.ndarray):
107109
for _i in range(repeat):

0 commit comments

Comments
 (0)