human
I wrote the below code

    @attach_name_and_doc(fun, argnum, 'Gradient')
    @add_error_hints
    def gradfun(*args, **kwargs):
        args = list(args)
        args[argnum] = safe_type(args[argnum])
        (vjp, ans) = make_vjp(fun, argnum)(*args, **kwargs)
        return vjp(vspace(ans).ones())
    return gradfun

What's a good function header?

Here's a good function header?
assistant
#Here's a good function header.

def grad(fun, argnum=0):