INNER CODE UNIT · Python

autotime

tskit-dev/tskit · python/benchmark/run.py:77

def autotime(setup, code):
    t = timeit.Timer(setup=setup, stmt=code)
    try:
        one_run = t.timeit(number=1)
    except Exception as e:
        print(f"{code}: Error running benchmark: {e}")
        return None
    num_trials = int(max(1, 2 / one_run))
    return one_run, num_trials, t.timeit(number=num_trials) / num_trials


def run_benchmarks(keyword_filter):
    results = {}
    for benchmark in tqdm.tqdm(config["benchmarks"]):
        bench_name = benchmark.get("name", benchmark["code"])
        if keyword_filter not in bench_name:
            continue
        params = benchmark.get("parameters", {"noop": [None]})

View source record →