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

villekuosmanen · aliberts · chrisheninger · commit 9cc59a098e03 · 2025-01-26T10:13:08.000-07:00
Co-authored-by: Simon Alibert &lt;75076266+aliberts@users.noreply.github.com&gt;
diff --git 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 = [

Original file line number	Diff line number	Diff line change
`@@ -177,7 +177,7 @@ def show_episode(dataset_namespace, dataset_name, episode_id, dataset=dataset, e`
`177`	`177`	`{"url": url_for("static", filename=video_path), "filename": video_path.parent.name}`
`178`	`178`	`for video_path in video_paths`
`179`	`179`	`]`
`180`		`- tasks = dataset.meta.episodes[0]["tasks"]`
	`180`	`+ tasks = dataset.meta.episodes[episode_id]["tasks"]`
`181`	`181`	`else:`
`182`	`182`	`video_keys = [key for key, ft in dataset.features.items() if ft["dtype"] == "video"]`
`183`	`183`	`videos_info = [`