hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

fix tests

Browse Source
This commit is contained in:
Remi Cadene
2024-03-06 13:55:12 +00:00
committed by Simon Alibert
parent c2c0ef9927
commit 524d29aa80
4 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_datasets.py
View File

@@ -6,12 +6,17 @@ from .utils import init_config
@pytest.mark.parametrize(
"env_name",
"env_name,dataset_id",
[
"simxarm",
"pusht",
# TODO(rcadene): simxarm is depreciated for now
# ("simxarm", "lift"),
("pusht", "pusht"),
("aloha", "sim_insertion_human"),
("aloha", "sim_insertion_scripted"),
("aloha", "sim_transfer_cube_human"),
("aloha", "sim_transfer_cube_scripted"),
],
)
def test_factory(env_name):
cfg = init_config(overrides=[f"env={env_name}"])
def test_factory(env_name, dataset_id):
cfg = init_config(overrides=[f"env={env_name}", f"env.task={dataset_id}"])
offline_buffer = make_offline_buffer(cfg)