added frees to get_ python functions

itrharrison · Aug 3, 2022 · 707e070 · 707e070
1 parent 468d794
commit 707e070
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