From a1b5d0faf257d4956b84d9db8d39138c38b300f9 Mon Sep 17 00:00:00 2001 From: Ville Kuosmanen Date: Thu, 9 Jan 2025 08:39:48 +0000 Subject: [PATCH] fix(visualise): use correct language description for each episode id (#604) Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> --- lerobot/scripts/visualize_dataset_html.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/visualize_dataset_html.py b/lerobot/scripts/visualize_dataset_html.py index ec6eca22..39b4c27d 100644 --- a/lerobot/scripts/visualize_dataset_html.py +++ b/lerobot/scripts/visualize_dataset_html.py @@ -177,7 +177,7 @@ def run_server( {"url": url_for("static", filename=video_path), "filename": video_path.parent.name} for video_path in video_paths ] - tasks = dataset.meta.episodes[0]["tasks"] + tasks = dataset.meta.episodes[episode_id]["tasks"] else: video_keys = [key for key, ft in dataset.features.items() if ft["dtype"] == "video"] videos_info = [