From b1b67d65503b46f7da04785656b145ee82cc0157 Mon Sep 17 00:00:00 2001 From: Ben Rowland Date: Sat, 31 Dec 2016 16:52:47 +0000 Subject: BUG: Fix apply_along_axis() for when func1d() returns a non-ndarray (#8426) * BUG: Closes issue #8419 Fixes issue in apply_along_axis() where func1d() returns a non ndarray * BUG: Fix apply_along_axis() when func1d() returns a non-ndarray Closes issue #8419. Fixes issue in apply_along_axis() where func1d() returns a non ndarray by calling asanyarray() on result. This commit fixes a too long line in the test case. --- numpy/lib/shape_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/lib/shape_base.py') diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index e580690d1..6e92aa041 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -127,7 +127,7 @@ def apply_along_axis(func1d, axis, arr, *args, **kwargs): ind[n] = 0 n -= 1 i.put(indlist, ind) - res = func1d(arr[tuple(i.tolist())], *args, **kwargs) + res = asanyarray(func1d(arr[tuple(i.tolist())], *args, **kwargs)) outarr[tuple(i.tolist())] = res k += 1 if res.shape == (): -- cgit v1.2.1