fix

xadupre · xadupre · commit ced1fe98a003 · 2024-08-30T14:33:33.000+02:00
diff --git a/onnx_array_api/ort/ort_profile.py b/onnx_array_api/ort/ort_profile.py
@@ -101,7 +101,9 @@ def ort_profile(
     if providers is None:
         providers = ["CPUExecutionProvider"]
     sess = InferenceSession(obj, sess_options, providers=providers, **kwargs)
-    first = next(feeds.values())
+    for v in feeds.values():
+        first = v
+        break
 
     if isinstance(first, numpy.ndarray):
         for _i in range(repeat):
