Skip to content

Commit 45950c7

Browse files
authored
Update candidateSelectorToXiPi.cxx
1 parent 748728f commit 45950c7

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

PWGHF/TableProducer/candidateSelectorToXiPi.cxx

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,12 @@
1414
/// \author Federica Zanone <federica.zanone@cern.ch>, Heidelberg University
1515
/// \author Tao Fang <tao.fang@cern.ch>, Central China Normal University
1616

17+
#include <string>
18+
#include <vector>
19+
1720
#include "PWGHF/Core/HfMlResponseXic0ToXiPi.h"
1821
#include "PWGHF/Core/SelectorCuts.h"
22+
1923
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
2024
#include "PWGHF/DataModel/CandidateSelectionTables.h"
2125
#include "PWGHF/Utils/utilsAnalysis.h"
@@ -41,8 +45,6 @@
4145

4246
#include <cstdint>
4347
#include <cstdlib>
44-
#include <string>
45-
#include <vector>
4648

4749
using namespace o2;
4850
using namespace o2::aod;
@@ -142,14 +144,14 @@ struct HfCandidateSelectorToXiPi {
142144
Configurable<LabeledArray<double>> cutsMl{"cutsMl", {hf_cuts_ml::Cuts[0], hf_cuts_ml::NBinsPt, hf_cuts_ml::NCutScores, hf_cuts_ml::labelsPt, hf_cuts_ml::labelsCutScore}, "ML selections per pT bin"};
143145
Configurable<int> nClassesMl{"nClassesMl", static_cast<int>(hf_cuts_ml::NCutScores), "Number of classes in ML model"};
144146
Configurable<std::vector<std::string>> namesInputFeatures{"namesInputFeatures", std::vector<std::string>{"feature1", "feature2"}, "Names of ML model input features"};
145-
147+
146148
// CCDB configuration
147149
Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
148150
Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{"EventFiltering/PWGHF/BDTXic0ToXipiKf"}, "Paths of models on CCDB"};
149151
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)"};
150152
Configurable<int64_t> timestampCCDB{"timestampCCDB", -1, "timestamp of the ONNX file for ML model used to query in CCDB"};
151153
Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"};
152-
154+
153155
o2::analysis::HfMlResponseXic0ToXiPi<float> hfMlResponse;
154156
std::vector<float> outputMlXic0ToXiPi = {};
155157
o2::ccdb::CcdbApi ccdbApi;
@@ -556,7 +558,7 @@ struct HfCandidateSelectorToXiPi {
556558
} else {
557559
registry.fill(HIST("hSelMassCharmBaryon"), 0);
558560
}
559-
561+
560562
// ML selections
561563
if (applyMl) {
562564
bool isSelectedMlXic0 = false;

0 commit comments

Comments
 (0)