Example Plot Scatter Jitter

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

from bencher.results.holoview_results.distribution_result.scatter_jitter_result import ScatterJitterResult
import bencher as bn

import random


class ScatterJitterDemo(bn.ParametrizedSweep):
    """Scatter with jitter across cache backends."""

    backend = bn.StringSweep(["redis", "memcached", "local"])
    noise_scale = bn.FloatSweep(default=0.0, bounds=[0.0, 1.0])

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

    def benchmark(self):
        lookup = {"redis": 1.2, "memcached": 0.9, "local": 0.3}
        self.distance = lookup[self.backend]
        if self.noise_scale > 0:
            self.distance += random.gauss(0, self.noise_scale)


def example_plot_scatter_jitter(run_cfg: bn.BenchRunCfg | None = None) -> bn.Bench:
    """Plot Type: Scatter Jitter."""
    bench = ScatterJitterDemo().to_bench(run_cfg)
    res = bench.plot_sweep(input_vars=["backend"], result_vars=["distance"], const_vars=dict(noise_scale=0.15))
    bench.report.append(res.to(ScatterJitterResult))

    return bench


if __name__ == "__main__":
    bn.run(example_plot_scatter_jitter, subsampling_divisions=3, repeats=10)