From a544a2927b67a413c08d05ea0f24bddd450f2e68 Mon Sep 17 00:00:00 2001
From: JaeYoonCHO <jaeyoon.cho@cern.ch>
Date: Sat, 12 Sep 2026 13:21:49 +0900
Subject: [PATCH] Refactor the ML table

---
 .../TableProducer/treeCreatorXicToXiPiPi.cxx  | 34 ++++++-------------
 1 file changed, 10 insertions(+), 24 deletions(-)

diff --git a/PWGHF/TableProducer/treeCreatorXicToXiPiPi.cxx b/PWGHF/TableProducer/treeCreatorXicToXiPiPi.cxx
index a80f2df5af5..b407f3b51be 100644
--- a/PWGHF/TableProducer/treeCreatorXicToXiPiPi.cxx
+++ b/PWGHF/TableProducer/treeCreatorXicToXiPiPi.cxx
@@ -197,27 +197,20 @@ DECLARE_SOA_TABLE(HfCandXicToXiPiPiLiteMLs, "AOD", "HFXICXI2PIMLITE",
                   full::CandidateSelFlag,
                   full::Y,
                   full::Eta,
-                  full::Phi,
                   full::P,
                   full::Pt,
                   full::M,
-                  hf_cand_xic_to_xi_pi_pi::InvMassXi,
-                  hf_cand_xic_to_xi_pi_pi::InvMassLambda,
                   full::DecayLength,
-                  full::DecayLengthXY,
                   full::Cpa,
-                  full::CpaXY,
-                  hf_cand_xic_to_xi_pi_pi::CpaXi,
-                  hf_cand_xic_to_xi_pi_pi::CpaXYXi,
-                  hf_cand_xic_to_xi_pi_pi::CpaLambda,
-                  hf_cand_xic_to_xi_pi_pi::CpaXYLambda,
-                  full::ImpactParameterXi,
-                  full::ImpactParameterNormalisedXi,
                   full::ImpactParameterPi0,
                   full::ImpactParameterNormalisedPi0,
                   full::ImpactParameterPi1,
                   full::ImpactParameterNormalisedPi1,
-                  full::MaxNormalisedDeltaIP,
+                  hf_cand_xic_to_xi_pi_pi::NSigTpcPiFromXicPlus0,
+                  hf_cand_xic_to_xi_pi_pi::NSigTpcPiFromXicPlus1,
+                  hf_cand_xic_to_xi_pi_pi::NSigTpcBachelorPi,
+                  hf_cand_xic_to_xi_pi_pi::NSigTpcPiFromLambda,
+                  hf_cand_xic_to_xi_pi_pi::NSigTpcPrFromLambda,
                   full::MlScoreBkg,
                   full::MlScorePrompt,
                   full::MlScoreNonPrompt);
@@ -702,27 +695,20 @@ struct HfTreeCreatorXicToXiPiPi {
         candidate.isSelXicToXiPiPi(),
         candidate.y(o2::constants::physics::MassXiCPlus),
         candidate.eta(),
-        candidate.phi(),
         candidate.p(),
         candidate.pt(),
         candidate.invMassXicPlus(),
-        candidate.invMassXi(),
-        candidate.invMassLambda(),
         candidate.decayLength(),
-        candidate.decayLengthXY(),
         candidate.cpa(),
-        candidate.cpaXY(),
-        candidate.cpaXi(),
-        candidate.cpaXYXi(),
-        candidate.cpaLambda(),
-        candidate.cpaXYLambda(),
-        candidate.impactParameter0(),
-        candidate.impactParameterNormalised0(),
         candidate.impactParameter1(),
         candidate.impactParameterNormalised1(),
         candidate.impactParameter2(),
         candidate.impactParameterNormalised2(),
-        candidate.maxNormalisedDeltaIP(),
+        candidate.nSigTpcPiFromXicPlus0(),
+        candidate.nSigTpcPiFromXicPlus1(),
+        candidate.nSigTpcBachelorPi(),
+        candidate.nSigTpcPiFromLambda(),
+        candidate.nSigTpcPrFromLambda(),
         mlScoreBkg,
         mlScorePrompt,
         mlScoreNonPrompt);