Merge pull request #1 from itrharrison/bug-memory_leaks

address memory leak: added frees to get_ python functions
itrharrison · Aug 3, 2022 · 5de3617 · 5de3617
2 parents 468d794 + 707e070
commit 5de3617
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/hi_class_public/python/classy.pyx b/hi_class_public/python/classy.pyx
@@ -1023,6 +1023,9 @@ cdef class Class:
             for index in range(timesteps):
                 background[names[i]][index] = data[index*number_of_titles+i]
 
+        free(titles)
+        free(data)
+
         return background
 
     def get_thermodynamics(self):
@@ -1058,6 +1061,9 @@ cdef class Class:
             for index in range(timesteps):
                 thermodynamics[names[i]][index] = data[index*number_of_titles+i]
 
+        free(titles)
+        free(data)
+
         return thermodynamics
 
     def get_primordial(self):
@@ -1094,6 +1100,9 @@ cdef class Class:
             for index in range(timesteps):
                 primordial[names[i]][index] = data[index*number_of_titles+i]
 
+        free(titles)
+        free(data)
+
         return primordial
 
 
@@ -1244,6 +1253,9 @@ cdef class Class:
             else:
                 spectra[ic_key] = tmpdict
 
+        free(titles)
+        free(data)
+
         return spectra