lib/utils/timer.py

from time import perf_counter
from contextlib import contextmanager

@contextmanager
def timer():
    start = perf_counter()
    yield lambda: perf_counter() - start