rename

sdpython · sdpython · Nov 13, 2023 · Nov 12, 2023 · Nov 12, 2023 · Nov 12, 2023
commit ad82c19ec08d43448c063202727df8b0db20f707
diff --git a/_unittests/ut_light_api/test_translate_classic.py b/_unittests/ut_light_api/test_translate_classic.py
@@ -35,7 +35,7 @@ def test_check_code(self):
         outputs.append(make_tensor_value_info("Y", TensorProto.FLOAT, shape=[]))
         graph = make_graph(
             nodes,
-            "noname",
+            "onename",
             inputs,
             outputs,
             initializers,
@@ -77,7 +77,7 @@ def test_exp(self):
         outputs.append(make_tensor_value_info('Y', TensorProto.FLOAT, shape=[]))
         graph = make_graph(
             nodes,
-            'noname',
+            'light_api',
             inputs,
             outputs,
             initializers,
@@ -161,7 +161,7 @@ def test_transpose(self):
             outputs.append(make_tensor_value_info('Y', TensorProto.FLOAT, shape=[]))
             graph = make_graph(
                 nodes,
-                'noname',
+                'light_api',
                 inputs,
                 outputs,
                 initializers,
@@ -223,7 +223,7 @@ def test_topk_reverse(self):
             outputs.append(make_tensor_value_info('Indices', TensorProto.FLOAT, shape=[]))
             graph = make_graph(
                 nodes,
-                'noname',
+                'light_api',
                 inputs,
                 outputs,
                 initializers,

diff --git a/onnx_array_api/light_api/inner_emitter.py b/onnx_array_api/light_api/inner_emitter.py
@@ -65,10 +65,11 @@ def _emit_begin_graph(self, **kwargs: Dict[str, Any]) -> List[str]:
         return lines
 
     def _emit_end_graph(self, **kwargs: Dict[str, Any]) -> List[str]:
+        name = kwargs.get("name", "noname")
         lines = [
             "graph = make_graph(",
             "    nodes,",
-            "    'noname',",
+            f"    {name!r},",
             "    inputs,",
             "    outputs,",
             "    initializers,",

diff --git a/onnx_array_api/light_api/translate.py b/onnx_array_api/light_api/translate.py
@@ -113,11 +113,16 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:
                         ),
                     )
                 )
+        if isinstance(self.proto_, (GraphProto, FunctionProto)):
+            name = self.proto_.name
+        else:
+            name = self.proto_.graph.name
         rows.extend(
             self.emitter(
                 EventType.END_FUNCTION
                 if isinstance(self.proto_, FunctionProto)
-                else EventType.END_GRAPH
+                else EventType.END_GRAPH,
+                name=name,
             )
         )