Function wrapped

dask/array/ufunc.py:19–24

(*args, **kwargs)

    """Wrap up numpy function into dask.array"""

    def wrapped(*args, **kwargs):
        dsk = [arg for arg in args if hasattr(arg, "_elemwise")]
        if len(dsk) > 0:
            return dsk[0]._elemwise(numpy_ufunc, *args, **kwargs)
        else:
            return numpy_ufunc(*args, **kwargs)

    wrapped.__name__ = numpy_ufunc.__name__
    return derived_from(source)(wrapped)