Aggregate works

2025-02-23 18:18:46 +00:00
parent e2e6f6e666
commit c36d2253d0
4 changed files with 106 additions and 26 deletions
--- a/tests/test_aggregate_datasets.py
+++ b/tests/test_aggregate_datasets.py
@@ -0,0 +1,19 @@
+from lerobot.common.datasets.aggregate import aggregate_datasets
+from tests.fixtures.constants import DUMMY_REPO_ID
+
+
+def test_aggregate_datasets(tmp_path, lerobot_dataset_factory):
+    dataset_0 = lerobot_dataset_factory(
+        root=tmp_path / "test_0",
+        repo_id=DUMMY_REPO_ID + "_0",
+        total_episodes=10,
+        total_frames=400,
+    )
+    dataset_1 = lerobot_dataset_factory(
+        root=tmp_path / "test_1",
+        repo_id=DUMMY_REPO_ID + "_1",
+        total_episodes=10,
+        total_frames=400,
+    )
+
+    dataset_2 = aggregate_datasets([dataset_0, dataset_1])