Skip to content

Commit d248107

Browse files
committed
Fix memory leak in _memory_usage
1 parent b15e80a commit d248107

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

stringdtype/stringdtype/src/dtype.c

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -246,8 +246,6 @@ stringdtype_new(PyTypeObject *NPY_UNUSED(cls), PyObject *args, PyObject *kwds)
246246
static void
247247
stringdtype_dealloc(StringDTypeObject *self)
248248
{
249-
// Need to deallocate all the memory allocated during setitem.
250-
251249
PyArrayDescr_Type.tp_dealloc((PyObject *)self);
252250
}
253251

stringdtype/stringdtype/src/main.c

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@ _memory_usage(PyObject *NPY_UNUSED(self), PyObject *obj)
6464

6565
} while (iternext(iter));
6666

67+
NpyIter_Deallocate(iter);
68+
6769
PyObject *ret = PyLong_FromSize_t(memory_usage);
6870

6971
return ret;

0 commit comments

Comments
 (0)