diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2015-09-27 12:33:09 -0600 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2015-09-27 12:33:09 -0600 |
| commit | ea289ee45dffa85d2fde07685eb3334816d8af7c (patch) | |
| tree | 92d94ad578e6d2ba26def28837725f6cf0e5690e /numpy/lib/shape_base.py | |
| parent | 41afcc3681d250f231aea9d9f428a9e197a47f6e (diff) | |
| parent | f29c387272a9279f82ab04bbbe1bb68040b6d383 (diff) | |
| download | numpy-ea289ee45dffa85d2fde07685eb3334816d8af7c.tar.gz | |
Merge pull request #6371 from seberg/pr-5771
BUG: Make sure warning for array split is always applied
Diffstat (limited to 'numpy/lib/shape_base.py')
| -rw-r--r-- | numpy/lib/shape_base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index 26c2aab04..b2beef0a8 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -424,7 +424,7 @@ def array_split(ary, indices_or_sections, axis=0): # This "kludge" was introduced here to replace arrays shaped (0, 10) # or similar with an array shaped (0,). # There seems no need for this, so give a FutureWarning to remove later. - if sub_arys[-1].size == 0 and sub_arys[-1].ndim != 1: + if any(arr.size == 0 and arr.ndim != 1 for arr in sub_arys): warnings.warn("in the future np.array_split will retain the shape of " "arrays with a zero size, instead of replacing them by " "`array([])`, which always has a shape of (0,).", |