WIP

lerobot/scripts/visualize_dataset.py

@@ -295,10 +295,10 @@ def write_episode_data_html(output_dir, file_name, js_fname, ep_index, dataset):
 def write_episodes_list_html(output_dir, file_name, ep_indices, ep_html_fnames, dataset):
     """Write an html file containing information related to the dataset and a list of links to
     html pages of episodes."""
-    soup, body = create_html_page(dataset.hf_dataset.info.dataset_name)
+    soup, body = create_html_page("TODO")
 
     h3 = soup.new_tag("h3")
-    h3.string = dataset.hf_dataset.info.dataset_name
+    h3.string = "TODO"
     body.append(h3)
 
     ul_info = soup.new_tag("ul")

tests/test_visualize_dataset.py

@@ -25,9 +25,8 @@ from lerobot.scripts.visualize_dataset import visualize_dataset
 def test_visualize_dataset(tmpdir, repo_id):
     rrd_path = visualize_dataset(
         repo_id,
-        episode_index=0,
-        batch_size=32,
-        save=True,
+        episode_indices=[0],
         output_dir=tmpdir,
+        serve=False,
     )
     assert rrd_path.exists()