Function load_variant

benchmarking/vllm/plot_tpot.py:62–68

(model_dir: Path, variant_key: str)

def latest_timestamp(variant_dir: Path) -> Path:
    dirs = sorted(d for d in variant_dir.iterdir() if d.is_dir() and (d / "summary.csv").exists())
    if not dirs:
        raise FileNotFoundError(f"No completed runs (with summary.csv) found in {variant_dir}")
    return dirs[-1]


def load_variant(model_dir: Path, variant_key: str) -> pd.DataFrame | None:
    variant_dir = model_dir / variant_key
    if not variant_dir.exists():
        print(f"Warning: {variant_dir} not found, skipping")