admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-15 00:29:52 +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 a225127527
commit 1f658023f1
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
@@ -28,6 +28,7 @@ from lerobot.scripts.lerobot_edit_dataset import (
RemoveFeatureConfig,
SplitConfig,
_validate_config,
TrimEpisodeStartConfig,
)
@@ -47,6 +48,7 @@ class TestOperationTypeParsing:
("merge", MergeConfig),
("remove_feature", RemoveFeatureConfig),
("modify_tasks", ModifyTasksConfig),
("trim_episode_start", TrimEpisodeStartConfig),
("convert_image_to_video", ConvertImageToVideoConfig),
("info", InfoConfig),
],
@@ -77,6 +79,7 @@ class TestOperationTypeParsing:
("merge", MergeConfig),
("remove_feature", RemoveFeatureConfig),
("modify_tasks", ModifyTasksConfig),
("trim_episode_start", TrimEpisodeStartConfig),
("convert_image_to_video", ConvertImageToVideoConfig),
("info", InfoConfig),
],