Skip to content

Commit af5d602

Browse files
authored
Update candidateSelectorXic0ToXiPiKf.cxx
1 parent 05ac90e commit af5d602

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

PWGHF/TableProducer/candidateSelectorXic0ToXiPiKf.cxx

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
#include "Common/Core/TrackSelection.h"
2525
#include "Common/Core/TrackSelectorPID.h"
2626

27-
#include "PWGHF/Core/HfMlResponseXic0ToXiPikf.h"
27+
#include "PWGHF/Core/HfMlResponseXic0ToXiPiKf.h"
2828
#include "PWGHF/Core/SelectorCuts.h"
2929

3030
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
@@ -46,7 +46,7 @@ enum PidInfoStored {
4646
/// Struct for applying Xic0 -> Xi pi selection cuts
4747
struct HfCandidateSelectorXic0ToXiPiKf {
4848
Produces<aod::HfSelToXiPiKf> hfSelToXiPi;
49-
Produces<aod::HfMlToXiPikf> hfMlToXiPi;
49+
Produces<aod::HfMlToXiPiKf> hfMlToXiPi;
5050

5151
// LF analysis selections
5252
Configurable<double> radiusCascMin{"radiusCascMin", 0.5, "Min cascade radius"};
@@ -134,13 +134,13 @@ struct HfCandidateSelectorXic0ToXiPiKf {
134134

135135
// CCDB configuration
136136
Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
137-
Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{"EventFiltering/PWGHF/BDTXic0ToXipikf"}, "Paths of models on CCDB"};
138-
Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"ModelHandler_onnx_Xic0ToXipikf.onnx"}, "ONNX file names for each pT bin (if not from CCDB full path)"};
137+
Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{"EventFiltering/PWGHF/BDTXic0ToXipiKf"}, "Paths of models on CCDB"};
138+
Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"ModelHandler_onnx_Xic0ToXipiKf.onnx"}, "ONNX file names for each pT bin (if not from CCDB full path)"};
139139
Configurable<int64_t> timestampCCDB{"timestampCCDB", -1, "timestamp of the ONNX file for ML model used to query in CCDB"};
140140
Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"};
141141

142-
o2::analysis::HfMlResponseXic0ToXiPikf<float> hfMlResponse;
143-
std::vector<float> outputMlXic0ToXiPikf = {};
142+
o2::analysis::HfMlResponseXic0ToXiPiKf<float> hfMlResponse;
143+
std::vector<float> outputMlXic0ToXiPiKf = {};
144144
o2::ccdb::CcdbApi ccdbApi;
145145

146146
TrackSelectorPr selectorProton;
@@ -548,11 +548,11 @@ struct HfCandidateSelectorXic0ToXiPiKf {
548548
if (applyMl) {
549549
bool isSelectedMlXic0 = false;
550550
std::vector<float> inputFeaturesXic0 = hfMlResponse.getInputFeatures(candidate, trackPiFromLam, trackPiFromCasc, trackPiFromCharm);
551-
isSelectedMlXic0 = hfMlResponse.isSelectedMl(inputFeaturesXic0, ptCand, outputMlXic0ToXiPikf);
551+
isSelectedMlXic0 = hfMlResponse.isSelectedMl(inputFeaturesXic0, ptCand, outputMlXic0ToXiPiKf);
552552
if (!isSelectedMlXic0) {
553553
continue;
554554
}
555-
hfMlToXiPi(outputMlXic0ToXiPikf);
555+
hfMlToXiPi(outputMlXic0ToXiPiKf);
556556
}
557557

558558
hfSelToXiPi(statusPidCharmBaryon, statusPidCascade, statusPidLambda, statusInvMassCharmBaryon, statusInvMassCascade, statusInvMassLambda, resultSelections, infoTpcStored, infoTofStored,

0 commit comments

Comments
 (0)