diff options
Diffstat (limited to 'numpy/lib/shape_base.py')
| -rw-r--r-- | numpy/lib/shape_base.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index 9c16982c4..da0b6a5b2 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -96,11 +96,10 @@ def apply_along_axis(func1d, axis, arr, *args, **kwargs): # handle negative axes arr = asanyarray(arr) nd = arr.ndim + if not (-nd <= axis < nd): + raise IndexError('axis {0} out of bounds [-{1}, {1})'.format(axis, nd)) if axis < 0: axis += nd - if axis >= nd: - raise ValueError("axis must be less than arr.ndim; axis=%d, rank=%d." - % (axis, nd)) # arr, with the iteration axis at the end in_dims = list(range(nd)) |