fix memory leak #131

Xtra-Computing · Mar 24, 2019 · 5946a70 · 5946a70
1 parent d71b447
commit 5946a70
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/thundersvmScikit.py b/python/thundersvmScikit.py
@@ -77,6 +77,10 @@ def __init__(self, kernel, degree,
         # if self.max_mem_size != -1:
         #     thundersvm.set_memory_size(c_void_p(self.model), self.max_mem_size)
 
+    def __del__(self):
+        if self.model is not None:
+            thundersvm.model_free(c_void_p(self.model))
+
     def label_validate(self, y):
 
         return column_or_1d(y, warn=True).astype(np.float64)