Timsty's picture
Add files using upload-large-folder tool
700dd75 verified
Raw
History Blame Contribute Delete
350 Bytes
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")