doc

sdpython · sdpython · Nov 13, 2023 · Nov 12, 2023 · Nov 12, 2023 · Nov 12, 2023
commit d75473f8d411087e4cc348427b89ef5fcaa224bc
diff --git a/_doc/api/light_api.rst b/_doc/api/light_api.rst
@@ -19,6 +19,12 @@ translate
 Classes for the Light API
 =========================
 
+ProtoType
++++++++++
+
+.. autoclass:: onnx_array_api.light_api.model.ProtoType
+    :members:
+
 OnnxGraph
 +++++++++
 

diff --git a/_unittests/ut_npx/test_npx.py b/_unittests/ut_npx/test_npx.py
@@ -1,5 +1,6 @@
 import inspect
 import unittest
+import sys
 from contextlib import redirect_stdout
 from io import StringIO
 
@@ -1355,6 +1356,7 @@ def test_clip_none(self):
         got = ref.run(None, {"A": x})
         self.assertEqualArray(y, got[0])
 
+    @unittest.skipIf(sys.platform == "win32", reason="unstable on windows")
     def test_arange_inline(self):
         # arange(5)
         f = arange_inline(Input("A"))

diff --git a/onnx_array_api/light_api/model.py b/onnx_array_api/light_api/model.py
@@ -24,6 +24,11 @@
 
 
 class ProtoType(IntEnum):
+    """
+    The same code can be used to output a GraphProto, a FunctionProto or a ModelProto.
+    This class specifies the output type at the beginning of the code.
+    """
+
     FUNCTION = 1
     GRAPH = 2
     MODEL = 3