Merge pull request #44 from aldro61/memory_leak

Memory leak. Solved issue #43
aldro61 · Apr 11, 2017 · 3206313 · 3206313
2 parents 209f908 + b1344b8
commit 3206313
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cpp_extensions/utility_python_bindings.cpp b/cpp_extensions/utility_python_bindings.cpp
@@ -160,7 +160,17 @@ find_max(PyObject *self, PyObject *args){
         opti_kinds_data[i] = (int) best_solution.best_feat_kind[i];
     }
 
+    if (feature_weights){
+        Py_DECREF(feature_weights);
+    }
+    else{
+        delete [] feature_weights_data;
+    }
 
+    Py_DECREF(X);
+    Py_DECREF(y);
+    Py_DECREF(X_argsort_by_feature);
+    Py_DECREF(example_idx);
 
     return Py_BuildValue("d,N,N,N",
                          opti_utility,