https://x.com/willmcgugan/status/1882397013700141190

timer_profile.py

import contextlib
from time import perf_counter
from typing import Generator


@contextlib.contextmanager
def timer(subject: str = "time") -> Generator[None, None, None]:
    start = perf_counter()
    yield
    elapsed = perf_counter() - start
    elapsed_ms = elapsed * 1000
    print(f"{subject} elapsed {elapsed_ms:.4f}ms")