BUG: set the needs_api variable in np.where

juliantaylor · juliantaylor · commit 33b6326e9e74 · 2014-01-17T18:43:18.000+01:00
diff --git a/numpy/core/src/multiarray/multiarraymodule.c b/numpy/core/src/multiarray/multiarraymodule.c
@@ -2764,6 +2764,8 @@ PyArray_Where(PyObject *condition, PyObject *x, PyObject *y)
             goto fail;
         }
 
+        needs_api = NpyIter_IterationNeedsAPI(iter);
+
         if (!needs_api) {
             NPY_BEGIN_THREADS_THRESHOLDED(NpyIter_GetIterSize(iter));
         }

Original file line number	Diff line number	Diff line change
`@@ -2764,6 +2764,8 @@ PyArray_Where(PyObject condition, PyObject x, PyObject *y)`
`2764`	`2764`	`goto fail;`
`2765`	`2765`	`}`
`2766`	`2766`
	`2767`	`+ needs_api = NpyIter_IterationNeedsAPI(iter);`
	`2768`	`+`
`2767`	`2769`	`if (!needs_api) {`
`2768`	`2770`	`NPY_BEGIN_THREADS_THRESHOLDED(NpyIter_GetIterSize(iter));`
`2769`	`2771`	`}`