We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 36f8d2d commit de3280dCopy full SHA for de3280d
numpy/_core/src/umath/ufunc_object.c
@@ -5941,7 +5941,6 @@ ufunc_at(PyUFuncObject *ufunc, PyObject *args)
5941
NPY_AUXDATA_FREE(auxdata);
5942
5943
Py_XDECREF(op2_array);
5944
- Py_XDECREF(iter);
5945
Py_XDECREF(iter2);
5946
for (int i = 0; i < nop; i++) {
5947
Py_XDECREF(operation_descrs[i]);
@@ -5957,9 +5956,12 @@ ufunc_at(PyUFuncObject *ufunc, PyObject *args)
5957
5956
if (PyArray_FLAGS(op1_array) & NPY_ARRAY_WRITEBACKIFCOPY) {
5958
PyArray_DiscardWritebackIfCopy(op1_array);
5959
}
+ // iter might own the array, so it must be decref'd second
5960
+ Py_XDECREF(iter);
5961
return NULL;
5962
5963
else {
5964
5965
Py_RETURN_NONE;
5966
5967
0 commit comments