fix(rollout): properly propagating video_files_size_in_mb to lerobot_dataset (#3470)

This commit is contained in:
Maxime Ellerbach
2026-04-27 17:25:37 +02:00
committed by GitHub
parent 7451813da6
commit bf5148c6e8
5 changed files with 22 additions and 4 deletions
+12
View File
@@ -416,6 +416,18 @@ def test_create_initial_counts_zero(tmp_path):
assert dataset.num_frames == 0
def test_create_propagates_video_files_size_in_mb(tmp_path):
"""video_files_size_in_mb passed to create() is reflected in the dataset metadata."""
dataset = LeRobotDataset.create(
repo_id=DUMMY_REPO_ID,
fps=DEFAULT_FPS,
features=SIMPLE_FEATURES,
root=tmp_path / "ds",
video_files_size_in_mb=42.0,
)
assert dataset.meta.video_files_size_in_mb == 42.0
def test_add_frame_works_in_write_mode(tmp_path):
"""add_frame() succeeds on a dataset created via create()."""
dataset = LeRobotDataset.create(