Skip to content

vllm.benchmarks.sweep.plot

Classes:

PlotBinner dataclass

Methods:

  • apply –

    Applies this binner to a DataFrame.

Source code in vllm/benchmarks/sweep/plot.py 
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
@dataclass
class PlotBinner:
    var: str
    bin_size: float

    @classmethod
    def parse_str(cls, s: str):
        for op_key in PLOT_BINNERS:
            if op_key in s:
                key, value = s.split(op_key)
                return PLOT_BINNERS[op_key](key, float(value.removeprefix(op_key)))
        else:
            raise ValueError(
                f"Invalid operator for plot binner '{s}'. "
                f"Valid operators are: {sorted(PLOT_BINNERS)}",
            )

    def apply(self, df: "pd.DataFrame") -> "pd.DataFrame":
        """Applies this binner to a DataFrame."""
        df = df.copy()
        df[self.var] = df[self.var] // self.bin_size * self.bin_size
        return df

apply(df)

Applies this binner to a DataFrame.

Source code in vllm/benchmarks/sweep/plot.py 
145
146
147
148
149
def apply(self, df: "pd.DataFrame") -> "pd.DataFrame":
    """Applies this binner to a DataFrame."""
    df = df.copy()
    df[self.var] = df[self.var] // self.bin_size * self.bin_size
    return df

PlotFilterBase dataclass

Bases: ABC

Methods:

  • apply –

    Applies this filter to a DataFrame.

Source code in vllm/benchmarks/sweep/plot.py 
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
@dataclass
class PlotFilterBase(ABC):
    var: str
    target: str

    @classmethod
    def parse_str(cls, s: str):
        for op_key in PLOT_FILTERS:
            if op_key in s:
                key, value = s.split(op_key)
                return PLOT_FILTERS[op_key](
                    key,
                    value.removeprefix(op_key).strip("'").strip('"'),
                )
        else:
            raise ValueError(
                f"Invalid operator for plot filter '{s}'. "
                f"Valid operators are: {sorted(PLOT_FILTERS)}",
            )

    @abstractmethod
    def apply(self, df: "pd.DataFrame") -> "pd.DataFrame":
        """Applies this filter to a DataFrame."""
        raise NotImplementedError

apply(df) abstractmethod

Applies this filter to a DataFrame.

Source code in vllm/benchmarks/sweep/plot.py 
44
45
46
47
@abstractmethod
def apply(self, df: "pd.DataFrame") -> "pd.DataFrame":
    """Applies this filter to a DataFrame."""
    raise NotImplementedError

_convert_inf_nan_strings(data)

Convert string values "inf", "-inf", and "nan" to their float equivalents.

This handles the case where JSON serialization represents inf/nan as strings.

Source code in vllm/benchmarks/sweep/plot.py 
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
def _convert_inf_nan_strings(data: list[dict[str, object]]) -> list[dict[str, object]]:
    """
    Convert string values "inf", "-inf", and "nan" to their float equivalents.

    This handles the case where JSON serialization represents inf/nan as strings.
    """
    converted_data = []
    for record in data:
        converted_record = {}
        for key, value in record.items():
            if isinstance(value, str):
                if value in ["inf", "-inf", "nan"]:
                    converted_record[key] = float(value)
                else:
                    converted_record[key] = value
            else:
                converted_record[key] = value
        converted_data.append(converted_record)
    return converted_data