Python3 Decorator for measure time of function's execution

time_it.py

import time

def time_it(func):
    def wrapper(*args,**kwargs):
        start_time = time.time()
        result =func(*args, **kwargs)
        end_time = time.time()
        print(f"Function {func.__name__} took {end_time - start_time:.4f} seconds")
        return result
    return wrapper

@time_it
def compute_square(n):
    return [i * i for i in range(n)]
 
compute_square(1234567)