You can pass it as a separate parameter, see my other comment below. In particular, you shouldn't look inside the `ufunc` object in these routines.