From ea9125d6d11db712ef8d5224102643be17408d05 Mon Sep 17 00:00:00 2001
From: Matthias Schroeder <matthias.schroeder@desy.de>
Date: Thu, 23 Feb 2017 18:33:49 +0100
Subject: [PATCH] FIX memory leak

---
 src/MEMClassifier.cc | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/MEMClassifier.cc b/src/MEMClassifier.cc
index d06629c..0b2bf68 100644
--- a/src/MEMClassifier.cc
+++ b/src/MEMClassifier.cc
@@ -467,8 +467,10 @@ TH3D* MEMClassifier::GetBTagPDF(const char* prefix, const char* flavour) {
 
 double MEMClassifier::GetJetBProbability(const char* prefix, const char* flavour, double pt, double eta, double bdisc) {
     TH3D* h = GetBTagPDF(prefix, flavour);
-    int i = h->FindBin(pt, std::abs(eta), bdisc);
-    return h->GetBinContent(i);
+    const int i = h->FindBin(pt, std::abs(eta), bdisc);
+    const double c = h->GetBinContent(i);
+    delete h;
+    return c;
 }
 
 MEM::JetProbability MEMClassifier::GetJetBProbabilities(
-- 
GitLab