admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-17 09:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

add trim_episode_start dataset edit operation

Browse Source 
Add a dataset edit operation to trim the first N seconds from episodes while rebuilding frame and episode indices and metadata consistently. Skip episodes that are too short to trim and cover parsing plus metadata invariants with focused tests.

Made-with: Cursor
This commit is contained in:
pepijn
2026-03-06 13:58:57 +00:00
parent 0394fae446
commit 6bbc24a991
4 changed files with 466 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/scripts/test_edit_dataset_parsing.py
+3
View File
@@ -26,6 +26,7 @@ from lerobot.scripts.lerobot_edit_dataset import (
OperationConfig,
RemoveFeatureConfig,
SplitConfig,
TrimEpisodeStartConfig,
)
@@ -45,6 +46,7 @@ class TestOperationTypeParsing:
("merge", MergeConfig),
("remove_feature", RemoveFeatureConfig),
("modify_tasks", ModifyTasksConfig),
("trim_episode_start", TrimEpisodeStartConfig),
("convert_image_to_video", ConvertImageToVideoConfig),
],
)
@@ -62,6 +64,7 @@ class TestOperationTypeParsing:
("merge", MergeConfig),
("remove_feature", RemoveFeatureConfig),
("modify_tasks", ModifyTasksConfig),
("trim_episode_start", TrimEpisodeStartConfig),
("convert_image_to_video", ConvertImageToVideoConfig),
],
)