Skip to content

Commit 06d94cc

Browse files
authored
[PWGHF] Remove unnecessary caching of feature names in HF track index skim creator (#17810)
1 parent f55863b commit 06d94cc

1 file changed

Lines changed: 0 additions & 9 deletions

File tree

PWGHF/TableProducer/trackIndexSkimCreator.cxx

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1695,9 +1695,6 @@ struct HfTrackIndexSkimCreator {
16951695
const std::vector<double> ptBinsMl{0., 1.e10};
16961696
const std::vector<int> cutDirMl{o2::cuts_ml::CutDirection::CutGreater, o2::cuts_ml::CutDirection::CutSmaller, o2::cuts_ml::CutDirection::CutSmaller};
16971697
const std::array<LabeledArray<double>, kN3ProngDecaysUsedMlForHfFilters> thresholdMlScore3Prongs{config.thresholdMlScoreDplusToPiKPi, config.thresholdMlScoreLcToPiKP, config.thresholdMlScoreDsToPiKK, config.thresholdMlScoreXicToPiKP};
1698-
const std::vector<std::string> inputFeatures2Prongs = {"ptProng0", "dcaXyProng0", "dcaZProng0", "ptProng1", "dcaXyProng1", "dcaZProng1"};
1699-
const std::vector<std::string> inputFeatures3Prongs = {"ptProng0", "dcaXyProng0", "dcaZProng0", "ptProng1", "dcaXyProng1", "dcaZProng1", "ptProng2", "dcaXyProng2", "dcaZProng2"};
1700-
const std::vector<std::string> inputFeatures3ProngsWithPid = {"ptProng0", "dcaXyProng0", "dcaZProng0", "ptProng1", "dcaXyProng1", "dcaZProng1", "ptProng2", "dcaXyProng2", "dcaZProng2", "tpcNSigmaPrProng0", "tpcNSigmaPrProng2", "tpcNSigmaPiProng0", "tpcNSigmaPiProng2", "tpcNSigmaKaProng1"};
17011698

17021699
// initialise 2-prong ML response
17031700
hfMlResponse2Prongs.configure(ptBinsMl, config.thresholdMlScoreD0ToKPi, cutDirMl, 3);
@@ -1707,7 +1704,6 @@ struct HfTrackIndexSkimCreator {
17071704
} else {
17081705
hfMlResponse2Prongs.setModelPathsLocal(onnxFileNames2Prongs);
17091706
}
1710-
hfMlResponse2Prongs.cacheInputFeaturesIndices(inputFeatures2Prongs);
17111707
hfMlResponse2Prongs.init();
17121708

17131709
// initialise 3-prong ML responses
@@ -1723,11 +1719,6 @@ struct HfTrackIndexSkimCreator {
17231719
} else {
17241720
hfMlResponse3Prongs[iDecay3P].setModelPathsLocal(onnxFileNames3Prongs[iDecay3P]);
17251721
}
1726-
if ((doprocess2And3ProngsWithPvRefitWithPidForHfFiltersBdt || doprocess2And3ProngsNoPvRefitWithPidForHfFiltersBdt) && iDecay3P == aod::hf_cand_3prong::DecayType::LcToPKPi) {
1727-
hfMlResponse3Prongs[iDecay3P].cacheInputFeaturesIndices(inputFeatures3ProngsWithPid);
1728-
} else {
1729-
hfMlResponse3Prongs[iDecay3P].cacheInputFeaturesIndices(inputFeatures3Prongs);
1730-
}
17311722
hfMlResponse3Prongs[iDecay3P].init();
17321723
}
17331724
}

0 commit comments

Comments
 (0)