From acf253dcf6d28ea45de0555b51768e02f74a0180 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Thu, 30 May 2024 17:21:18 +0100 Subject: [PATCH] the more the merrier --- tests/test_datasets.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 73e9d7bb..afcbb3ba 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -43,7 +43,11 @@ from lerobot.common.utils.utils import init_hydra_config, seeded_context from tests.utils import DEFAULT_CONFIG_PATH, DEVICE -@pytest.mark.parametrize("env_name, repo_id, policy_name", lerobot.env_dataset_policy_triplets) +@pytest.mark.parametrize( + "env_name, repo_id, policy_name", + lerobot.env_dataset_policy_triplets + + [("aloha", ["lerobot/aloha_sim_insertion_human", "lerobot/aloha_sim_transfer_cube_human"], "act")], +) def test_factory(env_name, repo_id, policy_name): """ Tests that: