diff options
Diffstat (limited to 'numpy/core/src/multiarraymodule.c')
-rw-r--r-- | numpy/core/src/multiarraymodule.c | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/numpy/core/src/multiarraymodule.c b/numpy/core/src/multiarraymodule.c index dbbc1593a..49bcfca69 100644 --- a/numpy/core/src/multiarraymodule.c +++ b/numpy/core/src/multiarraymodule.c @@ -4677,6 +4677,7 @@ PyArray_Empty(int nd, intp *dims, PyArray_Descr *type, int fortran) if (type->hasobject) { PyArray_FillObjectArray(ret, Py_None); + if (PyErr_Occurred()) {Py_DECREF(ret); return NULL;} } return (PyObject *)ret; } @@ -4802,6 +4803,7 @@ PyArray_Zeros(int nd, intp *dims, PyArray_Descr *type, int fortran) PyObject *zero = PyInt_FromLong(0); PyArray_FillObjectArray(ret, zero); Py_DECREF(zero); + if (PyErr_Occurred()) {Py_DECREF(ret); return NULL;} } else { n = PyArray_NBYTES(ret); |