WIP

small adjustment add rerun hot fix WIP
2024-05-17 02:05:44 +00:00
parent 495da6deaf
commit 5cb19d589f
3 changed files with 601 additions and 170 deletions
--- a/tests/test_visualize_dataset.py
+++ b/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()