wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix episodes.jsonl

This commit is contained in:
Simon Alibert 2024-10-18 12:03:29 +02:00
parent 91e8ce772b
commit e7355ba595
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/v2/convert_dataset_v1_to_v2.py
View File

@ -662,7 +662,7 @@ def convert_dataset(
# Episodes # Episodes
episodes = [ episodes = [
{"episode_index": ep_idx, "tasks": [tasks_by_episodes[ep_idx]], "length": episode_lengths[ep_idx]} {"episode_index": ep_idx, "tasks": tasks_by_episodes[ep_idx], "length": episode_lengths[ep_idx]}
for ep_idx in episode_indices for ep_idx in episode_indices
] ]
write_jsonlines(episodes, v20_dir / "meta" / "episodes.jsonl") write_jsonlines(episodes, v20_dir / "meta" / "episodes.jsonl")