Function resolve_model_dirs

benchmarking/vllm/plot_tpot.py:49–52

Find all model directories matching the model name, including -trialN suffixes.

(results_dir: Path, model: str)

def tp_from_dir(results_dir: Path) -> int:
    """Parse the TP value out of the results dir name (e.g. ...-tp2). Defaults to 1."""
    m = re.search(r"-tp(\d+)(?:[-_].*)?$", results_dir.name)
    return int(m.group(1)) if m else 1


def resolve_model_dirs(results_dir: Path, model: str) -> list[Path]: