You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
350 B
15 lines
350 B
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")
|