From 3ceef3bef1952f7feb45d0e578752fa43796b1d2 Mon Sep 17 00:00:00 2001 From: Qizhi Chen Date: Wed, 31 Dec 2025 15:40:35 +0800 Subject: [PATCH] sync robomind2lerobot init with latest lerobot (#79) --- robomind2lerobot/robomind_h5.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/robomind2lerobot/robomind_h5.py b/robomind2lerobot/robomind_h5.py index ef68024..512b801 100644 --- a/robomind2lerobot/robomind_h5.py +++ b/robomind2lerobot/robomind_h5.py @@ -100,7 +100,15 @@ class RoboMINDDataset(LeRobotDataset): obj.image_transforms = None obj.delta_timestamps = 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.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 def save_episode(self, split, action_config: dict, episode_data: dict | None = None) -> None: