Skip to content

docs(nnx): fill missing docstrings in variables/rnglib, fix helpers g… - #5545

Open
greekera1000 wants to merge 1 commit into
google:mainfrom
greekera1000:docs/nnx-variables-rnglib-docstrings
Open

docs(nnx): fill missing docstrings in variables/rnglib, fix helpers g…#5545
greekera1000 wants to merge 1 commit into
google:mainfrom
greekera1000:docs/nnx-variables-rnglib-docstrings

Conversation

@greekera1000

Copy link
Copy Markdown

What does this PR do?

More docstring fixes for the NNX API reference (part of #5161):

  • with_metadata had no docstring, so it rendered as a bare signature in the reference — added one with a working example.
  • RngStream inherited the generic Pytree docstring ("Base class for all NNX objects.") in the reference — added a proper one.
  • Fixed a "contained" → "containing" typo in the List and Dict docstrings.

Scope: docs + docstrings only, no behavioral changes. Doctests for the touched modules pass locally.

Relates to #5161.

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: [docs]: Improve docs for NNX API Reference #5161
  • The documentation and docstrings adhere to the documentation guidelines.
  • This change includes necessary high-coverage tests. (No quality testing = no merge!)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant