BUG: ufunc: Missed a small update for the unitless reduction case

numpy · Aug 27, 2011 · c3fafa4 · c3fafa4
1 parent 1848be6
commit c3fafa4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/numpy/core/src/multiarray/reduction.h b/numpy/core/src/multiarray/reduction.h
@@ -43,7 +43,7 @@ typedef int (PyArray_AssignReduceUnitFunc)(PyArrayObject *result,
  *              npy_intp count = *countptr;
  *
  *              // Skip any first-visit elements
- *              if (NpyIter_IsFirstVisit(iter, 1)) {
+ *              if (NpyIter_IsFirstVisit(iter, 0)) {
  *                  if (stride0 == 0) {
  *                      --count;
  *                      --skip_first_count;

diff --git a/numpy/core/src/umath/ufunc_object.c b/numpy/core/src/umath/ufunc_object.c
@@ -2881,7 +2881,7 @@ PyUFunc_Reduce(PyUFuncObject *self, PyArrayObject *arr, PyArrayObject *out,
                     npy_intp count = *count_ptr;
 
                     /* Skip any first-visit elements */
-                    if (NpyIter_IsFirstVisit(iter, 1)) {
+                    if (NpyIter_IsFirstVisit(iter, 0)) {
                         if (stride[0] == 0) {
                             --count;
                             --skip_first_count;
@@ -2933,7 +2933,7 @@ PyUFunc_Reduce(PyUFuncObject *self, PyArrayObject *arr, PyArrayObject *out,
                     npy_intp count = *count_ptr;
 
                     /* Skip any first-visit elements */
-                    if (NpyIter_IsFirstVisit(iter, 1)) {
+                    if (NpyIter_IsFirstVisit(iter, 0)) {
                         if (stride[0] == 0) {
                             --count;
                             --skip_first_count;