Add real robot devices and scripts to control real robot (#288)

Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
2024-07-15 17:43:10 +02:00
parent 5ffcb48a9a
commit 5e54e39795
32 changed files with 3068 additions and 46 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"],