Skip to content

[python] Simplify merged LeRobot source metadata - #11074

Merged
XiaoHongbo-Hope merged 3 commits into
apache:masterfrom
XiaoHongbo-Hope:codex/lerobot-source-metadata-json
Nov 10, 2026
Merged

XiaoHongbo-Hope merged 3 commits into
apache:masterfrom
XiaoHongbo-Hope:codex/lerobot-source-metadata-json

Conversation

@XiaoHongbo-Hope

@XiaoHongbo-Hope XiaoHongbo-Hope commented Nov 10, 2026 •

Copy link
Copy Markdown
Contributor

Purpose

Merged LeRobot imports currently archive source metadata in Arrow IPC binary columns. Store source metadata in a simpler, queryable __datasets table with six fields: index, name, info, tasks, subtasks, and stats. Task/subtask ID-to-text mappings use typed arrays of rows; info and stats contain the original source JSON objects.

Derive source episode/frame ranges and FPS from the original counts and FPS in info, removing duplicate range/FPS columns. Frame and Episode dataset_index values reference __datasets.index; the training reader preserves source selection, source-local Episode IDs, and mixed-FPS support, including reads of existing table groups using legacy metadata names or IPC archive columns.

Remove Episode schema/patch archives. Compatible Episode extension columns shared by every source remain typed columns; source-specific extension columns and original Arrow schemas are no longer archived. This PR is independent of #11070.

Tests

  • 155 LeRobot aggregate tests passed, including Ray parallel imports, unequal source sizes, source selection, mixed FPS, and legacy source metadata.
  • flake8 with the repository configuration and git diff --check passed.

@XiaoHongbo-Hope
XiaoHongbo-Hope force-pushed the codex/lerobot-source-metadata-json branch 2 times, most recently from 9441c33 to 2ea4e27 Compare November 10, 2026 15:56
@XiaoHongbo-Hope XiaoHongbo-Hope changed the title Store merged LeRobot source metadata as JSON Store merged LeRobot source metadata without IPC Nov 10, 2026
@XiaoHongbo-Hope XiaoHongbo-Hope changed the title Store merged LeRobot source metadata without IPC [python] Store merged LeRobot source metadata without IPC Nov 10, 2026
@XiaoHongbo-Hope
XiaoHongbo-Hope force-pushed the codex/lerobot-source-metadata-json branch from 2ea4e27 to 9bf9e1d Compare November 10, 2026 16:08
@Zouxxyy Zouxxyy changed the title [python] Store merged LeRobot source metadata without IPC [python] Simplify merged LeRobot source metadata Nov 10, 2026
@XiaoHongbo-Hope

Copy link
Copy Markdown
Contributor Author

+1

@XiaoHongbo-Hope
XiaoHongbo-Hope merged commit 8f6f39f into apache:master Nov 10, 2026
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.

2 participants