Skip to content

Commit b0923ab

Browse files
authored
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.
1 parent 7f70b78 commit b0923ab

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/lerobot/datasets/lerobot_dataset.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -825,6 +825,8 @@ def save_episode(self, episode_data: dict | None = None) -> None:
825825
"""
826826
if not episode_data:
827827
episode_buffer = self.episode_buffer
828+
else:
829+
episode_buffer = episode_data
828830

829831
validate_episode_buffer(episode_buffer, self.meta.total_episodes, self.features)
830832

0 commit comments

Comments
 (0)