| import pytest | |
| def pytest_addoption(parser): | |
| parser.addoption( | |
| "--tensorboard-log-dir", | |
| action="store", | |
| default="logs/Gr00t_TRL_loco/.tensorboard", | |
| help="Directory containing tensorboard logs", | |
| ) | |
| def tensorboard_log_dir(request): | |
| return request.config.getoption("--tensorboard-log-dir") | |