Repository navigation
[python] Support distributed LeRobot import with Ray - #10483
Draft
XiaoHongbo-Hope wants to merge 3 commits into
Draft
XiaoHongbo-Hope wants to merge 3 commits into
XiaoHongbo-Hope wants to merge 3 commits into
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Purpose
Allow LeRobot v3 imports to run across Ray workers instead of a single process:
Workers reuse the existing validation/writer path and prepare complete Episode groups. Episodes sharing video files stay together. The driver commits frames only after every task succeeds, then publishes companion metadata, the initial BTree and optional tags. Existing calls remain single-process.
Requires an initialized Ray cluster and shared source/target storage. Supports FileIO URIs and shared directories; Hub IDs remain on the single-process path. Tasks are not automatically retried. Publication across tables is not atomic; failed imports may leave partial tables and uncommitted files.
Tests
72 passed, 29 optional-dependency skips. Six new tests cover grouping, API validation, real Ray-worker image/video imports against serial results, tags and worker failure without a frame commit. Flake8 and
git diff --checkpassed.Validated on a local two-CPU Ray cluster (Ray 2.55.1); multi-node OSS performance has not been measured.