fix

2024-07-14 17:12:57 +02:00
parent d878ec27d5
commit 8b4a1c7a7f
3 changed files with 3 additions and 20 deletions
--- a/tests/test_visualize_dataset.py
+++ b/tests/test_visualize_dataset.py
@@ -20,21 +20,6 @@ import pytest
 from lerobot.scripts.visualize_dataset import visualize_dataset


-@pytest.mark.parametrize(
-    "repo_id",
-    ["lerobot/pusht"],
-)
-def test_visualize_dataset(tmpdir, repo_id):
-    rrd_path = visualize_dataset(
-        repo_id,
-        episode_index=0,
-        batch_size=32,
-        save=True,
-        output_dir=tmpdir,
-    )
-    assert rrd_path.exists()
-
-
@pytest.mark.parametrize(
    "repo_id",
    ["lerobot/pusht"],