Add dataset fixtures

This commit is contained in:
Simon Alibert
2024-10-31 13:46:46 +01:00
parent ee51f54cb5
commit ff84024ee9
2 changed files with 155 additions and 0 deletions

View File

@@ -23,6 +23,9 @@ from lerobot import available_cameras, available_motors, available_robots
from lerobot.common.utils.utils import init_hydra_config
from tests.utils import DEVICE, ROBOT_CONFIG_PATH_TEMPLATE, make_camera, make_motors_bus
# Import fixture modules as plugins
pytest_plugins = ["tests.fixtures.dataset"]
def pytest_collection_finish():
print(f"\nTesting with {DEVICE=}")