diff options
Diffstat (limited to 'numpy/lib/UserArray.py')
| -rw-r--r-- | numpy/lib/UserArray.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/numpy/lib/UserArray.py b/numpy/lib/UserArray.py index fefa1e0cd..243b22d70 100644 --- a/numpy/lib/UserArray.py +++ b/numpy/lib/UserArray.py @@ -182,6 +182,9 @@ class UserArray: if len(shape(a)) == 0: return a else: return self.__class__(a) + def __array_wrap__(self, *args): + return self.__class__(args[0]) + def __setattr__(self,attr,value): if attr=='shape': self.array.shape=value |