Decorator to use Line_Profiler from within a script

def profile_line(fn):
    def deco(func):
        def wrapper(*args, **kwargs):
            lp = LineProfiler()
            lp_wrapper = lp(func)
            ret_fun = lp_wrapper(*args, **kwargs)
            lp.print_stats()
            return ret_fun
        return wrapper
    return deco(fn)