File size: 350 Bytes
700dd75 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import pytest
def pytest_addoption(parser):
parser.addoption(
"--tensorboard-log-dir",
action="store",
default="logs/Gr00t_TRL_loco/.tensorboard",
help="Directory containing tensorboard logs",
)
@pytest.fixture
def tensorboard_log_dir(request):
return request.config.getoption("--tensorboard-log-dir")
|