Wrap up numpy function into dask.array
(numpy_ufunc, source=np)
| 17 | |
| 18 | |
| 19 | def wrap_elemwise(numpy_ufunc, source=np): |
| 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) |
| 31 | |
| 32 | |
| 33 | class da_frompyfunc: |
no test coverage detected
searching dependent graphs…