Function _plot_scatter_line

benchmarking/vllm/plot_tpot.py:132–169

Plot median lines with scatter dots for one or more series. Each entry in *series* is (label, dataframe). Colors are looked up from VARIANT_COLORS. The dataframe must contain *x_col* and *y_col* columns.

(
    ax: plt.Axes,
    series: list[tuple[str, pd.DataFrame]],
    x_col: str,
    y_col: str,
    ylabel: str,
    hline: float | None = None,
)

            1 - merged["median_tpot_ms_candidate"] / merged["median_tpot_ms_base"]
        ) * 100
        frames.append(merged[["max_concurrency", "speedup_pct"]])
    return pd.concat(frames, ignore_index=True) if frames else pd.DataFrame()


# ---------------------------------------------------------------------------
# Shared scatter+median line plot
# ---------------------------------------------------------------------------

FIGSIZE = (6.4, 5)
TPOT_FIGSIZE = (6.0, 5)
DOT_SIZE = 50
DOT_ALPHA = 0.4


def _plot_scatter_line(
    ax: plt.Axes,
    series: list[tuple[str, pd.DataFrame]],
    x_col: str,
    y_col: str,
    ylabel: str,
    hline: float | None = None,
):
    """Plot median lines with scatter dots for one or more series.

    Each entry in *series* is (label, dataframe). Colors are looked up from
    VARIANT_COLORS. The dataframe must contain *x_col* and *y_col* columns.
    """
    all_x = set()
    for label, sdf in series:
        color = VARIANT_COLORS[label]
        marker = VARIANT_MARKERS.get(label, "o")
        medians = sdf.groupby(x_col)[y_col].median()
        ax.plot(medians.index, medians.values, marker=marker, color=color, label=label, zorder=3)
        ax.scatter(
            sdf[x_col],
            sdf[y_col],
            color=color,
            alpha=DOT_ALPHA,
            s=DOT_SIZE,
            zorder=2,
            edgecolors="none",