diff --git a/src/lbench/cli/env.py b/src/lbench/cli/env.py index f227e14..19d40e6 100644 --- a/src/lbench/cli/env.py +++ b/src/lbench/cli/env.py @@ -9,7 +9,7 @@ def get_lbench_root_dir() -> Path: Resolve the lbench root directory. Priority: - 1. $LBENCH_ROOT_DIR if set + 1. $LBENCH_ROOT if set 2. ./lbench_results relative to CWD The resolved path is created if missing and written back diff --git a/src/lbench/pytest/plugin.py b/src/lbench/pytest/plugin.py index fd8c7c4..4e3fbbb 100644 --- a/src/lbench/pytest/plugin.py +++ b/src/lbench/pytest/plugin.py @@ -1,6 +1,7 @@ from __future__ import annotations from datetime import datetime +from pathlib import Path import pytest @@ -18,6 +19,7 @@ def pytest_addoption(parser): "--lbench-root", action="store", default=None, + type=Path, help="Root directory for lbench runs", ) @@ -29,6 +31,8 @@ def pytest_configure(config: pytest.Config): root = config.getoption("--lbench-root") if root is None: root = get_lbench_root_dir() + else: # Support paths like "~/results" + root = root.expanduser() run_id = datetime.now().strftime("%Y%m%d-%H%M%S") run_dir = root / run_id