import time from contextlib import contextmanager @contextmanager def timer(name: str): start = time.time() print(f"{name}...") yield print(f" -> {name} completed in {time.time() - start:.2f} sec")