Wrap a function.
(func_name)
| 15 | |
| 16 | |
| 17 | def wrap(func_name): |
| 18 | """ |
| 19 | Wrap a function. |
| 20 | """ |
| 21 | |
| 22 | def wrapped(self, *a, **kw): |
| 23 | a = getattr(self.arr, func_name)(*a, **kw) |
| 24 | return a if not isinstance(a, np.ndarray) else type(self)(a) |
| 25 | |
| 26 | return wrapped |
| 27 | |
| 28 | |
| 29 | def dispatch_property(prop_name): |