(*args, **kwargs)
| 20 | """Wrap up numpy function into dask.array""" |
| 21 | |
| 22 | def wrapped(*args, **kwargs): |
| 23 | dsk = [arg for arg in args if hasattr(arg, "_elemwise")] |
| 24 | if len(dsk) > 0: |
| 25 | return dsk[0]._elemwise(numpy_ufunc, *args, **kwargs) |
| 26 | else: |
| 27 | return numpy_ufunc(*args, **kwargs) |
| 28 | |
| 29 | wrapped.__name__ = numpy_ufunc.__name__ |
| 30 | return derived_from(source)(wrapped) |