Profile any Python function easily.

profile_function.py

from functools import wraps
import cProfile


def profileit(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        profile = cProfile.Profile()
        return_value = profile.runcall(func, *args, **kwargs)
        profile.dump_stats(f'{func.__name__}.profile')
        return return_value

    return wrapper
  
  # then read the output with snakeviz