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

fix(visualise): use correct language description for each episode id (#604) 

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
This commit is contained in:
Ville Kuosmanen 2025-01-09 08:39:48 +00:00 committed by GitHub
parent bc16e1b497
commit 5097cd900e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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/scripts/visualize_dataset_html.py
View File

@ -177,7 +177,7 @@ def run_server(
{"url": url_for("static", filename=video_path), "filename": video_path.parent.name} {"url": url_for("static", filename=video_path), "filename": video_path.parent.name}
for video_path in video_paths for video_path in video_paths
] ]
tasks = dataset.meta.episodes[0]["tasks"] tasks = dataset.meta.episodes[episode_id]["tasks"]
else: else:
video_keys = [key for key, ft in dataset.features.items() if ft["dtype"] == "video"] video_keys = [key for key, ft in dataset.features.items() if ft["dtype"] == "video"]
videos_info = [ videos_info = [