human
Here's a good function header given the following:
I wrote the below code

    double_val_fun = value_and_multigrad(fun, argnums=argnums)
    def multigrad_fun(*args, **kwargs):
        return double_val_fun(*args, **kwargs)[1]
    return multigrad_fun

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

def multigrad(fun, argnums=[0]):