hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

fix(dataset): Use provided episode_data in save_episode (#1740)

Browse Source 
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.
This commit is contained in:
lxk
2025-08-22 21:24:02 +08:00
committed by GitHub
parent 7f70b78f32
commit b0923ab74b
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/datasets/lerobot_dataset.py
View File

@@ -825,6 +825,8 @@ class LeRobotDataset(torch.utils.data.Dataset):
""" """
if not episode_data: if not episode_data:
episode_buffer = self.episode_buffer episode_buffer = self.episode_buffer
else:
episode_buffer = episode_data
validate_episode_buffer(episode_buffer, self.meta.total_episodes, self.features) validate_episode_buffer(episode_buffer, self.meta.total_episodes, self.features)