fix order

xadupre · xadupre · commit 092dfa23498e · 2024-04-01T11:28:20.000+02:00
diff --git a/_unittests/ut_translate_api/test_translate.py b/_unittests/ut_translate_api/test_translate.py
@@ -80,9 +80,9 @@ def test_transpose(self):
             """
             (
                 start(opset=19)
-                .vin('X', elem_type=TensorProto.FLOAT)
                 .cst(np.array([-1, 1], dtype=np.int64))
                 .rename('r')
+                .vin('X', elem_type=TensorProto.FLOAT)
                 .bring('X', 'r')
                 .Reshape()
                 .rename('r0_0')
@@ -166,9 +166,9 @@ def test_export_if(self):
             f"""
             (
                 start(opset=19)
-                .vin('X', elem_type=TensorProto.FLOAT)
                 .cst(np.array([0.0], dtype=np.float32))
                 .rename('r')
+                .vin('X', elem_type=TensorProto.FLOAT)
                 .bring('X')
                 .ReduceSum(keepdims=1, noop_with_empty_axes=0)
                 .rename('Xs')
@@ -202,9 +202,9 @@ def test_aionnxml(self):
             """
             (
                 start(opset=19, opsets={'ai.onnx.ml': 3})
-                .vin('X', elem_type=TensorProto.FLOAT)
                 .cst(np.array([-1, 1], dtype=np.int64))
                 .rename('r')
+                .vin('X', elem_type=TensorProto.FLOAT)
                 .bring('X', 'r')
                 .Reshape()
                 .rename('USE')
diff --git a/_unittests/ut_translate_api/test_translate_builder.py b/_unittests/ut_translate_api/test_translate_builder.py
@@ -65,6 +65,32 @@ def light_api(
         got = ref.run(None, {"X": a})[0]
         self.assertEqualArray(np.exp(a), got)
 
+    def test_zdoc(self):
+        onx = (
+            start()
+            .vin("X")
+            .reshape((-1, 1))
+            .Transpose(perm=[1, 0])
+            .rename("Y")
+            .vout()
+            .to_onnx()
+        )
+        code = translate(onx, api="builder")
+        expected = dedent(
+            """
+            (
+                start()
+                .vin("X")
+                .reshape((-1, 1))
+                .Transpose(perm=[1, 0])
+                .rename("Y")
+                .vout()
+                .to_onnx()
+            )"""
+        ).strip("\n")
+        self.maxDiff = None
+        self.assertEqual(expected, code)
+
 
 if __name__ == "__main__":
     unittest.main(verbosity=2)
diff --git a/onnx_array_api/translate_api/translate.py b/onnx_array_api/translate_api/translate.py
@@ -82,6 +82,16 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:
                 self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.graph.name)
             )
 
+        for i in initializers:
+            rows.extend(
+                self.emitter(
+                    EventType.INITIALIZER,
+                    name=i.name,
+                    init=i,
+                    value=to_array_extended(i),
+                )
+            )
+
         rows.extend(self.emitter(EventType.BEGIN_SIGNATURE))
 
         for i in inputs:
@@ -107,16 +117,6 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:
 
         rows.extend(self.emitter(EventType.END_SIGNATURE))
 
-        for i in initializers:
-            rows.extend(
-                self.emitter(
-                    EventType.INITIALIZER,
-                    name=i.name,
-                    init=i,
-                    value=to_array_extended(i),
-                )
-            )
-
         for node in nodes:
             atts = self.extract_attributes(node)
             rows.extend(
