admin/any4lerobot
1
0
Fork 0
mirror of https://github.com/Tavish9/any4lerobot.git synced 2026-05-23 09:49:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

sync robomind2lerobot init with latest lerobot (#79)

Browse Source
This commit is contained in:
Qizhi Chen
2025-12-31 15:40:35 +08:00
committed by GitHub
parent ca4b740d9a
commit 3ceef3bef1
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
robomind2lerobot/robomind_h5.py
+8
View File
@@ -100,7 +100,15 @@ class RoboMINDDataset(LeRobotDataset):
obj.image_transforms = None obj.image_transforms = None
obj.delta_timestamps = None obj.delta_timestamps = None
obj.delta_indices = None obj.delta_indices = None
obj._absolute_to_relative_idx = None
obj.video_backend = video_backend if video_backend is not None else get_safe_default_codec() obj.video_backend = video_backend if video_backend is not None else get_safe_default_codec()
obj.writer = None
obj.latest_episode = None
obj._current_file_start_frame = None
# Initialize tracking for incremental recording
obj._lazy_loading = False
obj._recorded_frames = 0
obj._writer_closed_for_reading = False
return obj return obj
def save_episode(self, split, action_config: dict, episode_data: dict | None = None) -> None: def save_episode(self, split, action_config: dict, episode_data: dict | None = None) -> None: