From e4bafb2a4a3c682aaec18cc184c6a95ee1a30c26 Mon Sep 17 00:00:00 2001 From: Sebastian Berg Date: Mon, 3 Sep 2012 16:22:06 +0200 Subject: Reset flags when Axes are removed. Array might now be 1D, or removed axes of size 1 destroy contiguousity. --- numpy/core/src/multiarray/shape.c | 3 +++ 1 file changed, 3 insertions(+) (limited to 'numpy') diff --git a/numpy/core/src/multiarray/shape.c b/numpy/core/src/multiarray/shape.c index 067232632..684d42713 100644 --- a/numpy/core/src/multiarray/shape.c +++ b/numpy/core/src/multiarray/shape.c @@ -1186,4 +1186,7 @@ PyArray_RemoveAxesInPlace(PyArrayObject *arr, npy_bool *flags) /* The final number of dimensions */ fa->nd = idim_out; + + /* Update contiguous flags */ + PyArray_UpdateFlags(arr, NPY_ARRAY_C_CONTIGUOUS | NPY_ARRAY_F_CONTIGUOUS); } -- cgit v1.2.1