20 lines
586 B
Python
20 lines
586 B
Python
|
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])
|