fix doc

xadupre · xadupre · commit 64cdc471cea1 · 2024-03-01T15:40:37.000+01:00
diff --git a/_doc/tutorial/onnx_api.rst b/_doc/tutorial/onnx_api.rst
@@ -71,7 +71,11 @@ the true implementation would be the following.
             n2 = oh.make_node("Pow", ["dxy", "two"], ["dxy2"])
             n3 = oh.make_node("ReduceSum", ["dxy2"], [output_name])
             graph = oh.make_graph([n1, n2, n3], "euclidian", [X, Y], [Z], [two])
-            model = oh.make_model(graph, opset_imports=[oh.make_opsetid("", opset)])
+            model = oh.make_model(
+                graph,
+                opset_imports=[oh.make_opsetid("", opset)],
+                ir_version=9,
+            )
             return model
 
 
diff --git a/onnx_array_api/plotting/graphviz_helper.py b/onnx_array_api/plotting/graphviz_helper.py
@@ -203,6 +203,7 @@ def plot_dot(
 
         import matplotlib.pyplot as plt
         import onnx.parser
+        from onnx_array_api.plotting.graphviz_helper import plot_dot
 
         model = onnx.parser.parse_model(
                     '''
@@ -212,7 +213,8 @@ def plot_dot(
                         four = Add(two, two)
                         z = Mul(four, four)
                     }''')
-        ax = plot_dot(dot)
+
+        ax = plot_dot(model)
         ax.set_title("Dummy graph")
         plt.show()
     """
diff --git a/onnx_array_api/reference/evaluator_yield.py b/onnx_array_api/reference/evaluator_yield.py
@@ -240,7 +240,7 @@ def enumerate_summarized(
         :param feed_inputs: dictionary `{ input name: input value }`
         :param raise_exc: raises an exception if the execution fails or stop
             where it is
-        :param keep_tensor:keep the tensor in order to compute precise distances
+        :param keep_tensor: keep the tensor in order to compute precise distances
         :return: iterator on ResultExecution
         """
         for kind, name, value, op_type in self.enumerate_results(
diff --git a/onnx_array_api/validation/docs.py b/onnx_array_api/validation/docs.py
@@ -30,7 +30,9 @@ def make_euclidean(
     n2 = oh.make_node("Pow", ["dxy", "two"], ["dxy2"])
     n3 = oh.make_node("ReduceSum", ["dxy2"], [output_name])
     graph = oh.make_graph([n1, n2, n3], "euclidian", [X, Y], [Z], [two])
-    model = oh.make_model(graph, opset_imports=[oh.make_opsetid("", opset)])
+    model = oh.make_model(
+        graph, opset_imports=[oh.make_opsetid("", opset)], ir_version=9
+    )
     return model
 
 
