fix unit test

Author: xadupre

_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')

_unittests/ut_translate_api/test_translate_classic.py

@@ -138,13 +138,13 @@ def test_transpose(self):
             initializers = []
             sparse_initializers = []
             functions = []
+            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             initializers.append(
                 from_array(
                     np.array([-1, 1], dtype=np.int64),
                     name='r'
                 )
             )
-            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             nodes.append(
                 make_node_extended(
                     'Reshape',
@@ -278,13 +278,13 @@ def test_aionnxml(self):
             initializers = []
             sparse_initializers = []
             functions = []
+            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             initializers.append(
                 from_array(
                     np.array([-1, 1], dtype=np.int64),
                     name='r'
                 )
             )
-            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             nodes.append(
                 make_node_extended(
                     'Reshape',

onnx_array_api/translate_api/translate.py

@@ -75,6 +75,8 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:
                     domain=self.proto_.domain,
                 )
             )
+        elif isinstance(self.proto_, GraphProto):
+            rows.extend(self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.name))
         else:
             rows.extend(
                 self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.graph.name)