Decorator for line profiler

line_profiler_decorator.py

def profile(func):
    import functools
    @functools.wraps(func)
    def inner(*args, **kwargs):
        from line_profiler import LineProfiler
        lp = LineProfiler()
        lp_wrapper = lp(func)
        result = lp_wrapper(*args, **kwargs)
        lp.print_stats()
        return result
    return inner