lerobot/tests/utils.py

7 lines
177 B
Python
Raw Normal View History

import os
2024-03-28 02:33:48 +08:00
# Pass this as the first argument to init_hydra_config.
DEFAULT_CONFIG_PATH = "lerobot/configs/default.yaml"
DEVICE = os.environ.get('LEROBOT_TESTS_DEVICE', "cuda")