[DistPart] Fix corner case in dist partition which always led to an assertion error being triggered.#7395

Merged

Rhett-Ying merged 3 commits intodmlc:masterfrom thvasilo:dist-part-dimension-error

May 14, 2024

Commits on May 10, 2024

Fix corner case which always led to an assertion error being triggered.
thvasilo
committed

Commits on May 13, 2024

Address review comments.
thvasilo
committed

Commits on May 14, 2024

Apply lintrunner changes
thvasilo
committed