From 9898417d2af4019004828805db5f07b776181663 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Wed, 6 Mar 2019 22:33:28 -0500 Subject: Fix some memory leaks --- gensvm/cython_wrapper/wrapper.pyx | 2 ++ 1 file changed, 2 insertions(+) (limited to 'gensvm') diff --git a/gensvm/cython_wrapper/wrapper.pyx b/gensvm/cython_wrapper/wrapper.pyx index f98341b..0858103 100644 --- a/gensvm/cython_wrapper/wrapper.pyx +++ b/gensvm/cython_wrapper/wrapper.pyx @@ -82,6 +82,7 @@ def train_wrap( error_msg = check_model(model) if error_msg: gensvm_free_model(model) + gensvm_free_model(seed_model) free_data(data) error_repl = error_msg.decode('utf-8') raise ValueError(error_repl) @@ -106,6 +107,7 @@ def train_wrap( # free model and data gensvm_free_model(model); + gensvm_free_model(seed_model) free_data(data); return (V, n_SV, iter_count, training_error, fit_status) -- cgit v1.2.3