better error message

sdpython · sdpython · Jun 5, 2023 · Mar 20, 2023 · Jun 3, 2023 · Jun 3, 2023
commit 8068ea9611097c6a4a5e8dbc9d9d80d8c24c6ec2
diff --git a/CHANGELOGS.rst b/CHANGELOGS.rst
@@ -0,0 +1,7 @@
+Change Logs
+===========
+
+0.2.0
++++++
+
+* :pr:`3`: fixes Array API with onnxruntime
diff --git a/_doc/index.rst b/_doc/index.rst
@@ -34,6 +34,7 @@ well as to execute it.
     tutorial/index
     api/index
     auto_examples/index
+    ../CHANGELOGS
 
 Sources available on
 `github/onnx-array-api <https://github.com/sdpython/onnx-array-api>`_,

diff --git a/onnx_array_api/npx/npx_jit_eager.py b/onnx_array_api/npx/npx_jit_eager.py
@@ -367,6 +367,9 @@ def jit_call(self, *values, **kwargs):
         try:
             res = fct.run(*values)
         except Exception as e:
+            from ..plotting.text_plot import onnx_simple_text_plot
+
+            text = onnx_simple_text_plot(self.onxs[key])
             raise RuntimeError(
                 f"Unable to run function for key={key!r}, "
                 f"types={[type(x) for x in values]}, "
@@ -375,7 +378,7 @@ def jit_call(self, *values, **kwargs):
                 f"kwargs={kwargs}, "
                 f"self.input_to_kwargs_={self.input_to_kwargs_}, "
                 f"f={self.f} from module {self.f.__module__!r} "
-                f"onnx={self.onxs[key]}."
+                f"onnx=\n---\n{text}\n---\n{self.onxs[key]}"
             ) from e
         self.info("-", "jit_call")
         return res