From 2067fa1f9216b983544e5690c62ed2ddf62672c6 Mon Sep 17 00:00:00 2001 From: Ville Kuosmanen Date: Mon, 30 Dec 2024 20:15:32 +0000 Subject: [PATCH] fix(visualise): use correct language description for each episode id --- 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 ec6eca22f8..39b4c27d6a 100644 --- a/lerobot/scripts/visualize_dataset_html.py +++ b/lerobot/scripts/visualize_dataset_html.py @@ -177,7 +177,7 @@ def show_episode(dataset_namespace, dataset_name, episode_id, dataset=dataset, e {"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 = [