Merge branch 'main' of https://github.com/sdpython/onnx-array-api

xadupre · xadupre · commit ace236c4855f · 2024-04-08T23:57:29.000+02:00
diff --git a/onnx_array_api/_command_lines_parser.py b/onnx_array_api/_command_lines_parser.py
@@ -123,8 +123,14 @@ def _cmd_compare(argv: List[Any]):
 
     parser = get_parser_compare()
     args = parser.parse_args(argv[1:])
+    if args.verbose in ("1", 1, "True", True):
+        print(f"[compare] first model {args.model1!r}")
+        print(f"[compare] second model {args.model2!r}")
     onx1 = onnx.load(args.model1)
     onx2 = onnx.load(args.model2)
+    if args.verbose in ("1", 1, "True", True):
+        print(f"[compare] first model has {len(onx1.graph.node)} nodes")
+        print(f"[compare] second model has {len(onx2.graph.node)} nodes")
     res1, res2, align, dc = compare_onnx_execution(
         onx1,
         onx2,
diff --git a/onnx_array_api/reference/evaluator_yield.py b/onnx_array_api/reference/evaluator_yield.py
@@ -642,15 +642,16 @@ def compare_onnx_execution(
             print("[compare_onnx_execution] loading first model")
         proto1 = load(model1) if isinstance(model1, str) else model1
         if verbose:
-            print("[compare_onnx_execution] loading first model")
+            print("[compare_onnx_execution] loading second model")
         proto2 = load(model2) if isinstance(model2, str) else model2
         res1 = list(_enumerate_result_no_execution(proto1))
         res2 = list(_enumerate_result_no_execution(proto2))
     else:
         return
 
     if verbose:
-        print(f"[compare_onnx_execution] got {len(res2)} results")
+        print(f"[compare_onnx_execution] got {len(res1)} results (first model)")
+        print(f"[compare_onnx_execution] got {len(res2)} results (second model)")
         print("[compare_onnx_execution] compute edit distance")
     dc = DistanceExecution()
     _, align = dc.distance_sequence(res1, res2)
