Skip to content

Instantly share code, notes, and snippets.

@richard-to

richard-to/timer.py

Created July 12, 2024 16:54
Download ZIP

Revisions

  1. richard-to created this gist Jul 12, 2024.
    8 changes: 8 additions & 0 deletions timer.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,8 @@
    @contextmanager
    def timer(name):
    """Context manager for timing code blocks with custom name."""
    start_time = time.time()
    yield
    end_time = time.time()
    elapsed_time = (end_time - start_time) * 1000
    print(f"{name} took {elapsed_time:.2f} ms")