improves documentation

Author: xadupre

_doc/examples/plot_export_locate_issue.py

@@ -103,5 +103,6 @@ def forward(self, x: torch.Tensor, ys: list[torch.Tensor]):
 #   File "onnx-diagnostic/_doc/examples/plot_export_locate_issue.py", line 25, in forward
 #       z = x * caty
 
+# %%
 
-doc.plot_legend("was inferred to be a constant", "torch.export.export", "tomato")
+doc.plot_legend("was inferred\nto be a constant", "torch.export.export", "tomato")

_doc/examples/plot_export_with_dynamic_shapes_auto.py

@@ -94,4 +94,4 @@ def forward(self, x, y, z):
 
 # %%
 
-doc.plot_legend("dynamic shapes inferred", "torch.export.export", "tomato")
+doc.plot_legend("dynamic shapes\ninferred", "torch.export.export", "tomato")

_doc/examples/plot_failing_onnxruntime_evaluator.py

@@ -106,4 +106,4 @@
 # It is possible to insert prints in the python code to print
 # more information or debug if needed.
 
-doc.plot_legend("onnxruntime running step by step", "OnnxruntimeEvaluator", "lightgrey")
+doc.plot_legend("onnxruntime running\nstep by step", "OnnxruntimeEvaluator", "lightgrey")

_doc/examples/plot_failing_reference_evaluator.py

@@ -81,4 +81,4 @@
 # It is possible to insert prints in the python code to print
 # more information or debug if needed.
 
-doc.plot_legend("Python Runtime for ONNX", "ExtendedReferenceEvalutor", "lightgrey")
+doc.plot_legend("Python Runtime\nfor ONNX", "ExtendedReferenceEvalutor", "lightgrey")

onnx_diagnostic/doc.py

@@ -12,7 +12,7 @@ def plot_legend(
 ) -> "matplotlib.axes.Axes":  # noqa: F821
     import matplotlib.pyplot as plt
 
-    fig = plt.figure()
+    fig = plt.figure(figsize=(2, 2))
     ax = fig.add_subplot()
     ax.axis([0, 5, 0, 5])
     ax.text(2.5, 4, "END", fontsize=50, horizontalalignment="center")