ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 10:51:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(bug) in save_episode_data

Browse Source
This commit is contained in:
Michel Aractingi
2025-09-06 00:17:46 +02:00
parent 0747afdba7
commit 952f455446
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1063,8 +1063,10 @@ class LeRobotDataset(torch.utils.data.Dataset):
else:
df.to_parquet(path)
new_hf_dataset = Dataset.from_pandas(df)
self.hf_dataset = concatenate_datasets([self.hf_dataset, new_hf_dataset])
self.hf_dataset = (
concatenate_datasets([self.hf_dataset, ep_dataset]) if self.hf_dataset is not None else ep_dataset
)
self.hf_dataset.set_transform(hf_transform_to_torch)
metadata = {
"data/chunk_index": chunk_idx,