diff options
Diffstat (limited to 'numpy/core/fromnumeric.py')
| -rw-r--r-- | numpy/core/fromnumeric.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/numpy/core/fromnumeric.py b/numpy/core/fromnumeric.py index 38cf0aabf..98291f411 100644 --- a/numpy/core/fromnumeric.py +++ b/numpy/core/fromnumeric.py @@ -239,11 +239,7 @@ def nonzero(a): res = _wrapit(a, 'nonzero') else: res = nonzero() - - if len(res) == 1: - return res[0] - else: - raise ValueError, "Input argument must be 1d" + return res def shape(a): |