From 86058e26bc60b8ffd98304b5bcd7a20f0e6ef32e Mon Sep 17 00:00:00 2001
From: Fabrizio Grosa <fabrizio.grosa@cern.ch>
Date: Wed, 3 Aug 2022 19:21:26 +0200
Subject: [PATCH] Tag as bad candidates for which BDT inference fails

---
 EventFiltering/PWGHF/HFFilter.cxx | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/EventFiltering/PWGHF/HFFilter.cxx b/EventFiltering/PWGHF/HFFilter.cxx
index 7780c4595f3..d91d2c40042 100644
--- a/EventFiltering/PWGHF/HFFilter.cxx
+++ b/EventFiltering/PWGHF/HFFilter.cxx
@@ -711,6 +711,8 @@ struct HfFilter { // Main struct for HF triggers
 
       // apply ML models
       if (applyML && onnxFiles[kD0] != "") {
+        isCharmTagged = false;
+        isBeautyTagged = false;
         // TODO: add more feature configurations
         std::vector<Ort::Value> inputTensorD0;
         std::vector<float> inputFeaturesD0{trackPos.pt(), trackPos.dcaXY(), trackPos.dcaZ(), trackNeg.pt(), trackNeg.dcaXY(), trackNeg.dcaZ()};
@@ -845,6 +847,10 @@ struct HfFilter { // Main struct for HF triggers
 
       // apply ML models
       if (applyML) {
+        for (auto iCharmPart{0}; iCharmPart < kNCharmParticles - 1; ++iCharmPart) {
+          isCharmTagged[iCharmPart] = false;
+          isBeautyTagged[iCharmPart] = false;
+        }
         // TODO: add more feature configurations
         std::vector<float> inputFeatures{trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ()};
         std::vector<double> inputFeaturesD{trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ()};