Use the method

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

src/onnx_ir/external_data_test.py

@@ -173,13 +173,13 @@ def _simple_model(self) -> ir.Model:
         node_1 = ir.Node(
             "",
             "Op_1",
-            inputs=[node_0.outputs[0]],
+            inputs=[node_0.o()],
             num_outputs=1,
             name="node_1",
         )
         graph = ir.Graph(
             inputs=node_0.inputs,  # type: ignore
-            outputs=[node_1.outputs[0]],
+            outputs=[node_1.o()],
             initializers=[
                 ir.Value(name="tensor1", const_value=tensor1),
                 ir.Value(name="tensor2", const_value=tensor2),

src/onnx_ir/passes/common/clear_metadata_and_docstring_test.py

@@ -30,7 +30,7 @@ def test_pass_with_clear_metadata_and_docstring(self):
         )
         mul_node = ir.node(
             "Mul",
-            inputs=[add_node.outputs[0], inputs[1]],
+            inputs=[add_node.o(), inputs[1]],
             num_outputs=1,
             metadata_props={"mul_key": "mul_value"},
             doc_string="This is a Mul node",
@@ -69,7 +69,7 @@ def test_pass_with_clear_metadata_and_docstring(self):
         )
         sub_node = ir.node(
             "Sub",
-            inputs=[function.outputs[0], const_node.outputs[0]],
+            inputs=[function.o(), const_node.o()],
             num_outputs=1,
             metadata_props={"sub_key": "sub_value"},
             doc_string="This is a Sub node",

src/onnx_ir/passes/common/constant_manipulation.py

@@ -41,7 +41,7 @@ def call(self, model: ir.Model) -> ir.passes.PassResult:
             assert node.graph is not None
             if node.op_type != "Constant" or node.domain not in ("", "onnx.ai"):
                 continue
-            if node.outputs[0].is_graph_output():
+            if node.o().is_graph_output():
                 logger.debug(
                     "Constant node '%s' is used as output, so it can't be lifted.", node.name
                 )
@@ -54,7 +54,7 @@ def call(self, model: ir.Model) -> ir.passes.PassResult:
                 continue
 
             attr_name, attr_value = next(iter(node.attributes.items()))
-            initializer_name = node.outputs[0].name
+            initializer_name = node.o().name
             assert initializer_name is not None
             assert isinstance(attr_value, ir.Attr)
             tensor = self._constant_node_attribute_to_tensor(
@@ -73,7 +73,7 @@ def call(self, model: ir.Model) -> ir.passes.PassResult:
             assert node.graph is not None
             node.graph.register_initializer(initializer)
             # Replace the constant node with the initializer
-            ir.convenience.replace_all_uses_with(node.outputs[0], initializer)
+            ir.convenience.replace_all_uses_with(node.o(), initializer)
             node.graph.remove(node, safe=True)
             count += 1
             logger.debug(

src/onnx_ir/passes/common/constant_manipulation_test.py

@@ -36,8 +36,8 @@ def test_pass_with_lifting_float_and_int_constants_to_initializers(
         const_node = ir.node(
             "Constant", inputs=[], attributes={"value": constant_tensor}, num_outputs=1
         )
-        add_node = ir.node("Add", inputs=[inputs[0], const_node.outputs[0]])
-        mul_node = ir.node("Mul", inputs=[add_node.outputs[0], inputs[1]])
+        add_node = ir.node("Add", inputs=[inputs[0], const_node.o()])
+        mul_node = ir.node("Mul", inputs=[add_node.o(), inputs[1]])
 
         model = ir.Model(
             graph=ir.Graph(
@@ -92,21 +92,21 @@ def test_pass_with_lifting_constants_to_initializers_within_subgraph(
         )
         # then branch adds the constant to the input
         # else branch multiplies the input by the constant
-        add_node = ir.node("Add", inputs=[input_value, then_const_node.outputs[0]])
+        add_node = ir.node("Add", inputs=[input_value, then_const_node.o()])
         then_graph = ir.Graph(
             inputs=[],
-            outputs=[add_node.outputs[0]],
+            outputs=[add_node.o()],
             nodes=[then_const_node, add_node],
             opset_imports={"": 20},
         )
         else_constant_tensor = ir.tensor(np.random.rand(2, 3).astype(np.float32))
         else_const_node = ir.node(
             "Constant", inputs=[], attributes={"value": else_constant_tensor}, num_outputs=1
         )
-        mul_node = ir.node("Mul", inputs=[input_value, else_const_node.outputs[0]])
+        mul_node = ir.node("Mul", inputs=[input_value, else_const_node.o()])
         else_graph = ir.Graph(
             inputs=[],
-            outputs=[mul_node.outputs[0]],
+            outputs=[mul_node.o()],
             nodes=[else_const_node, mul_node],
             opset_imports={"": 20},
         )
@@ -179,14 +179,14 @@ def test_pass_with_lifting_constants_to_initializers_with_floats_ints_strings(
             num_outputs=1,
         )
         identity_node_constant = ir.node(
-            "Identity", inputs=[const_node.outputs[0]], num_outputs=1
+            "Identity", inputs=[const_node.o()], num_outputs=1
         )
         identity_node_input = ir.node("Identity", inputs=[input_value], num_outputs=1)
 
         model = ir.Model(
             graph=ir.Graph(
                 inputs=[input_value],
-                outputs=[identity_node_input.outputs[0], identity_node_constant.outputs[0]],
+                outputs=[identity_node_input.o(), identity_node_constant.o()],
                 nodes=[identity_node_input, const_node, identity_node_constant],
                 opset_imports={"": 20},
             ),
@@ -232,7 +232,7 @@ def test_not_lifting_constants_to_initializers_when_it_is_output(self):
         model = ir.Model(
             graph=ir.Graph(
                 inputs=[input_value],
-                outputs=[identity_node_input.outputs[0], const_node.outputs[0]],
+                outputs=[identity_node_input.o(), const_node.o()],
                 nodes=[identity_node_input, const_node],
                 opset_imports={"": 20},
             ),
@@ -272,7 +272,7 @@ def test_pass_with_lifting_constants_to_initializers_within_subgraph(
         add_node = ir.node("Add", inputs=[input_value, then_initializer_value])
         then_graph = ir.Graph(
             inputs=[],
-            outputs=[add_node.outputs[0]],
+            outputs=[add_node.o()],
             nodes=[add_node],
             opset_imports={"": 20},
             initializers=[then_initializer_value],
@@ -287,7 +287,7 @@ def test_pass_with_lifting_constants_to_initializers_within_subgraph(
         mul_node = ir.node("Mul", inputs=[input_value, else_initializer_value])
         else_graph = ir.Graph(
             inputs=[],
-            outputs=[mul_node.outputs[0]],
+            outputs=[mul_node.o()],
             nodes=[mul_node],
             opset_imports={"": 20},
             initializers=[else_initializer_value],
@@ -358,7 +358,7 @@ def test_pass_does_not_lift_initialized_inputs_in_subgraph(
             # The initializer is also an input. We don't lift it to the main graph
             # to preserve the graph signature
             inputs=[then_initializer_value],
-            outputs=[add_node.outputs[0]],
+            outputs=[add_node.o()],
             nodes=[add_node],
             opset_imports={"": 20},
             initializers=[then_initializer_value],
@@ -373,7 +373,7 @@ def test_pass_does_not_lift_initialized_inputs_in_subgraph(
         mul_node = ir.node("Mul", inputs=[input_value, else_initializer_value])
         else_graph = ir.Graph(
             inputs=[],
-            outputs=[mul_node.outputs[0]],
+            outputs=[mul_node.o()],
             nodes=[mul_node],
             opset_imports={"": 20},
             initializers=[else_initializer_value],