admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-18 02:00:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix metadata info.json

Browse Source
This commit is contained in:
Pepijn
2026-02-05 17:09:01 +01:00
parent 0753415244
commit a07ff640bb
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/port_datasets/slurm_mirror_dataset.py
+4 -3
View File
@@ -346,9 +346,10 @@ class MirrorDataAndMetadata(PipelineStep):
df.to_parquet(dst_parquet, index=False)
new_meta.info.update({
"total_episodes": dataset.meta.total_episodes,
"total_frames": dataset.meta.total_frames,
"total_tasks": dataset.meta.total_tasks,
"total_episodes": dataset.meta.info["total_episodes"],
"total_frames": dataset.meta.info["total_frames"],
"total_tasks": dataset.meta.info["total_tasks"],
"splits": dataset.meta.info.get("splits", {}),
})
write_info(new_meta.info, new_meta.root)