How to profile functions with decorators

profile_function.py

import cProfile
import pstats
import io


def profile(fnc):
    def inner(*arg, **kwargs):
        pr = cProfile.Profile()
        pr.enable()
        retval = fnc(*arg, **kwargs)
        pr.disable()
        s = io.StringIO()
        sortby = 'cumulative'
        ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
        ps.print_stats()
        print(s.getvalue())
        return retval

    return inner