Wrap up numpy function into dask.array
(numpy_ufunc, source=np)
| 14 | |
| 15 | |
| 16 | def wrap_elemwise(numpy_ufunc, source=np): |
| 17 | """Wrap up numpy function into dask.array""" |
| 18 | |
| 19 | def wrapped(*args, **kwargs): |
| 20 | dsk = [arg for arg in args if hasattr(arg, "_elemwise")] |
| 21 | if len(dsk) > 0: |
| 22 | return dsk[0]._elemwise(numpy_ufunc, *args, **kwargs) |
| 23 | else: |
| 24 | return numpy_ufunc(*args, **kwargs) |
| 25 | |
| 26 | # functools.wraps cannot wrap ufunc in Python 2.x |
| 27 | wrapped.__name__ = numpy_ufunc.__name__ |
| 28 | return derived_from(source)(wrapped) |
| 29 | |
| 30 | |
| 31 | class da_frompyfunc: |
no test coverage detected