Measure execution time of a function

stopwatch.py

def stopwatch():
    def decorator(func):
        def wrapper(*args, **kwargs):
            start_time = time.time()
            result = func(*args, **kwargs)
            end_time = time.time()
            elapsed_time = end_time - start_time
            print(f"time taken {elapsed_time: 2f}")
            return result

        return wrapper

    return decorator