From b0923ab74b7fb7ed688ef2abbe79607f3dee390a Mon Sep 17 00:00:00 2001 From: lxk <53181378+lxk-221@users.noreply.github.com> Date: Fri, 22 Aug 2025 21:24:02 +0800 Subject: [PATCH] fix(dataset): Use provided episode_data in save_episode (#1740) The 'episode_data' parameter was previously ignored, causing an error if provided. This change ensures it is correctly used, which allows for asynchronous episode saving by passing a copy of the episode buffer, preventing conflicts with the main data collection loop. --- src/lerobot/datasets/lerobot_dataset.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 617ac297f..a869cb920 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -825,6 +825,8 @@ class LeRobotDataset(torch.utils.data.Dataset): """ if not episode_data: episode_buffer = self.episode_buffer + else: + episode_buffer = episode_data validate_episode_buffer(episode_buffer, self.meta.total_episodes, self.features)