Skip to content

Adding set_mode_info#5071

Merged
copybara-service[bot] merged 1 commit into
google:mainfrom
chapman20j:set_mode_info
Nov 17, 2025
Merged

Adding set_mode_info#5071
copybara-service[bot] merged 1 commit into
google:mainfrom
chapman20j:set_mode_info

Conversation

@chapman20j

Copy link
Copy Markdown
Collaborator

What does this PR do?

Implements set_mode_info which displays information about the set_mode methods in a module.
This function recursively checks if nnx.Modules contain a set_mode attribute and collects the classes.
It then produces a string that contains information from the set_mode method signatures and docstrings from each module containing a set_mode method.

Fixes # (issue)

Checklist

  • This PR fixes a minor issue (e.g.: typo or small bug) or improves the docs (you can dismiss the other checks if that's the case).
  • This change is discussed in a Github issue/discussion (please add a link).
  • [x ] The documentation and docstrings adhere to the documentation guidelines.
  • [x ] This change includes necessary high-coverage tests. (No quality testing = no merge!)

@chapman20j chapman20j force-pushed the set_mode_info branch 6 times, most recently from 0c7e15e to 12f1c15 Compare November 4, 2025 18:32
@copybara-service copybara-service Bot merged commit 0b1d841 into google:main Nov 17, 2025
18 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants