Example Plot Scatter

"""Auto-generated example: Plot Type: Scatter."""

import bencher as bn

class ThroughputCompare(bn.ParametrizedSweep):
    """Throughput comparison across backends."""

    backend = bn.StringSweep(["redis", "memcached", "local"])

    distance = bn.ResultFloat("m", doc="Throughput distance metric")

    def benchmark(self):
        lookup = {"redis": 5.4, "memcached": 4.1, "local": 8.7}
        self.distance = lookup[self.backend]


def example_plot_scatter(run_cfg: bn.BenchRunCfg | None = None) -> bn.Bench:
    """Plot Type: Scatter."""
    bench = ThroughputCompare().to_bench(run_cfg)
    res = bench.plot_sweep(input_vars=["backend"], result_vars=["distance"])
    bench.report.append(res.to_scatter())

    return bench


if __name__ == "__main__":
    bn.run(example_plot_scatter, subsampling_divisions=3)