Skip to content

[Train] Fix all of Train's pydoclint issues#63365

Merged
matthewdeng merged 3 commits into
ray-project:masterfrom
pseudo-rnd-thoughts:fix-train-docstrings
May 18, 2026
Merged

[Train] Fix all of Train's pydoclint issues#63365
matthewdeng merged 3 commits into
ray-project:masterfrom
pseudo-rnd-thoughts:fix-train-docstrings

Conversation

@pseudo-rnd-thoughts
Copy link
Copy Markdown
Member

@pseudo-rnd-thoughts pseudo-rnd-thoughts commented May 15, 2026

Description

#52974 added pydoclint to pre-commit without fixing any issues through adding all the problematic docstrings to an ignore list.
However this means that all the docstrings that do have issues / problems with them that aren't raised or fixed.

This PR removes all of Train's ignores then uses Claude to fix all the docstrings / type hints (and reviewed by me to confirm implementations).

Signed-off-by: Mark Towers <mark@anyscale.com>
@pseudo-rnd-thoughts pseudo-rnd-thoughts requested a review from a team as a code owner May 15, 2026 11:22
@pseudo-rnd-thoughts pseudo-rnd-thoughts added the train Ray Train Related Issue label May 15, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request focuses on resolving a large number of pydoclint violations across the ray.train module by standardizing docstrings and type hints. Key improvements include adding missing Args, Returns, and Yields sections, reordering docstring parameters to match function signatures, and refining type annotations for clarity. A review comment correctly identifies a discrepancy in BackendExecutor._create_rank_world_size_mappings, where the docstring describes the return value as a tuple while the implementation and type hint specify a list.

Comment thread python/ray/train/_internal/backend_executor.py
Mark Towers added 2 commits May 15, 2026 12:38
Signed-off-by: Mark Towers <mark@anyscale.com>
@pseudo-rnd-thoughts pseudo-rnd-thoughts added the go add ONLY when ready to merge, run all tests label May 15, 2026
Copy link
Copy Markdown
Contributor

@matthewdeng matthewdeng left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.


@matthewdeng matthewdeng merged commit a8d5df5 into ray-project:master May 18, 2026
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

go add ONLY when ready to merge, run all tests train Ray Train Related Issue

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants