diff options
| -rw-r--r-- | numpy/core/src/multiarray/compiled_base.c | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/numpy/core/src/multiarray/compiled_base.c b/numpy/core/src/multiarray/compiled_base.c index 2f308c7d3..14f4a8f65 100644 --- a/numpy/core/src/multiarray/compiled_base.c +++ b/numpy/core/src/multiarray/compiled_base.c @@ -1441,6 +1441,9 @@ arr_add_docstring(PyObject *NPY_UNUSED(dummy), PyObject *args) } docstr = PyUnicode_AsUTF8(str); + if (docstr == NULL) { + return NULL; + } #else if (!PyArg_ParseTuple(args, "OO!:add_docstring", &obj, &PyString_Type, &str)) { return NULL; |