admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 01:41:54 +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
src/lerobot/datasets/lerobot_dataset.py
+4 -2
View File
@@ -1063,8 +1063,10 @@ class LeRobotDataset(torch.utils.data.Dataset):
else: else:
df.to_parquet(path) df.to_parquet(path)
new_hf_dataset = Dataset.from_pandas(df) self.hf_dataset = (
self.hf_dataset = concatenate_datasets([self.hf_dataset, new_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 = { metadata = {
"data/chunk_index": chunk_idx, "data/chunk_index": chunk_idx,