Implement all_nodes() and subgraphs() as convenience methods

[justinchuby]

Suggested by @xiaoyu-work, this PR created convenience methods all_nodes() and subgraphs() for easier access to the nested nodes and subgraphs. They are created so that it is easier to discover the methods.

[Copilot]

Pull Request Overview

This PR implements two convenience methods, all_nodes() and subgraphs(), to simplify traversals of nodes and subgraphs in both Graph and Function objects.

• Added all_nodes() and subgraphs() to the Graph class for nested node access.
• Modified graphs() in the model to yield the main graph and its subgraphs.
• Added analogous all_nodes() and subgraphs() methods to the Function class leveraging its underlying _graph.

Comments suppressed due to low confidence (3)

src/onnx_ir/_core.py:2385

• The subgraphs() method in Graph compares node.graph with self, while in Function it compares with self._graph. Consider adding a comment to clarify this intentional difference to aid API consistency.

if graph is self:

src/onnx_ir/_core.py:2905

• [nitpick] Since the graphs() method now yields self.graph before yielding subgraphs, consider updating the docstring to explain the intentional order, so that consumers of this API are clear on its behavior.

yield self.graph

src/onnx_ir/_core.py:3056

• In the Function class, subgraphs() checks if node.graph equals self._graph instead of self. Confirm and document this difference in comparison to the Graph implementation to ensure clarity.

if graph is self._graph:

[codecov]

Codecov Report

Attention: Patch coverage is 54.16667% with 11 lines in your changes missing coverage. Please review.

Project coverage is 73.35%. Comparing base (24096c6) to head (4e02e5e).

✅ All tests successful. No failed tests found.

Files with missing lines	Patch %	Lines
src/onnx_ir/_core.py	54.16%	10 Missing and 1 partial ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #76      +/-   ##
==========================================
- Coverage   73.48%   73.35%   -0.14%     
==========================================
  Files          37       37              
  Lines        4507     4526      +19     
  Branches      906      910       +4     
==========================================
+ Hits         3312     3320       +8     
- Misses        865      875      +10     
- Partials      330      331       +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[justinchuby]

@codecov-ai-reviewer test

[codecov-ai]

On it! Codecov is generating unit tests for this PR.

[justinchuby]

@codecov-ai-reviewer test

[codecov-ai]

On it! Codecov is generating unit tests for this PR.

[codecov-ai]

Sentry has determined that unit tests are not necessary for this PR.