Skip to content

Commit a60600b

Browse files
committed
Please consider the following formatting changes
1 parent 80d94fc commit a60600b

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

PWGHF/Core/HfMlResponseXicToXiPikf.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ class HfMlResponseXicToXiPikf : public HfMlResponse<TypeOutputScore>
8080
/// \param candidate is the Xic candidate
8181
/// \return inputFeatures vector
8282
template <typename T1, typename T2, typename T3>
83-
//std::vector<float> getInputFeatures(T1 const& candidate)
83+
// std::vector<float> getInputFeatures(T1 const& candidate)
8484
std::vector<float> getInputFeatures(T1 const& candidate, T2 const& lamProngPi, T2 const& cascProngPi, T3 const& charmBaryonProngPi)
8585
{
8686
std::vector<float> inputFeatures;

PWGHF/TableProducer/candidateSelectorXic0ToXiPiKf.cxx

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -123,26 +123,26 @@ struct HfCandidateSelectorXic0ToXiPiKf {
123123
Configurable<int> nClustersItsMin{"nClustersItsMin", 3, "Minimum number of ITS clusters requirement for pi <- charm baryon"};
124124
Configurable<int> nClustersItsInnBarrMin{"nClustersItsInnBarrMin", 1, "Minimum number of ITS clusters in inner barrel requirement for pi <- charm baryon"};
125125
Configurable<float> itsChi2PerClusterMax{"itsChi2PerClusterMax", 36, "Maximum value of chi2 fit over ITS clusters for pi <- charm baryon"};
126-
126+
127127
// ML inference
128128
Configurable<bool> applyMl{"applyMl", true, "Flag to apply ML selections"};
129129
Configurable<std::vector<double>> binsPtMl{"binsPtMl", std::vector<double>{hf_cuts_ml::vecBinsPt}, "pT bin limits for ML application"};
130130
Configurable<std::vector<int>> cutDirMl{"cutDirMl", std::vector<int>{hf_cuts_ml::vecCutDir}, "Whether to reject score values greater or smaller than the threshold"};
131131
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"};
132-
Configurable<int> nClassesMl{"nClassesMl", static_cast<int>(hf_cuts_ml::NCutScores), "Number of classes in ML model"};
132+
Configurable<int> nClassesMl{"nClassesMl", static_cast<int>(hf_cuts_ml::NCutScores), "Number of classes in ML model"};
133133
Configurable<std::vector<std::string>> namesInputFeatures{"namesInputFeatures", std::vector<std::string>{"feature1", "feature2"}, "Names of ML model input features"};
134-
134+
135135
// CCDB configuration
136136
Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
137137
Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{"EventFiltering/PWGHF/BDTXic"}, "Paths of models on CCDB"};
138138
Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"ModelHandler_onnx_XicToXipikf.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"};
141-
141+
142142
o2::analysis::HfMlResponseXicToXiPikf<float> hfMlResponse;
143143
std::vector<float> outputMlXicToXiPi = {};
144144
o2::ccdb::CcdbApi ccdbApi;
145-
145+
146146
TrackSelectorPr selectorProton;
147147
TrackSelectorPi selectorPion;
148148

@@ -203,7 +203,7 @@ struct HfCandidateSelectorXic0ToXiPiKf {
203203
registry.add("hSelMassCharmBaryon", "hSelMassCharmBaryon;status;entries", {HistType::kTH1D, {axisSel}});
204204
registry.add("hSelDcaXYToPvV0Daughters", "hSelDcaXYToPvV0Daughters;status;entries", {HistType::kTH1D, {axisSel}});
205205
registry.add("hSelDcaXYToPvPiFromCasc", "hSelDcaXYToPvPiFromCasc;status;entries", {HistType::kTH1D, {axisSel}});
206-
206+
207207
if (applyMl) {
208208
hfMlResponse.configure(binsPtMl, cutsMl, cutDirMl, nClassesMl);
209209
if (loadModelsFromCCDB) {
@@ -215,7 +215,6 @@ struct HfCandidateSelectorXic0ToXiPiKf {
215215
hfMlResponse.cacheInputFeaturesIndices(namesInputFeatures);
216216
hfMlResponse.init();
217217
}
218-
219218
}
220219

221220
void process(aod::HfCandToXiPiKf const& candidates,
@@ -225,7 +224,7 @@ struct HfCandidateSelectorXic0ToXiPiKf {
225224

226225
// looping over charm baryon candidates
227226
for (const auto& candidate : candidates) {
228-
227+
229228
auto ptCand = RecoDecay::sqrtSumOfSquares(candidate.pxCharmBaryon(), candidate.pyCharmBaryon());
230229

231230
bool resultSelections = true; // True if the candidate passes all the selections, False otherwise
@@ -556,7 +555,7 @@ struct HfCandidateSelectorXic0ToXiPiKf {
556555
continue;
557556
}
558557
}
559-
558+
560559
hfSelToXiPi(statusPidCharmBaryon, statusPidCascade, statusPidLambda, statusInvMassCharmBaryon, statusInvMassCascade, statusInvMassLambda, resultSelections, infoTpcStored, infoTofStored,
561560
trackPiFromCharm.tpcNSigmaPi(), trackPiFromCasc.tpcNSigmaPi(), trackPiFromLam.tpcNSigmaPi(), trackPrFromLam.tpcNSigmaPr(),
562561
trackPiFromCharm.tofNSigmaPi(), trackPiFromCasc.tofNSigmaPi(), trackPiFromLam.tofNSigmaPi(), trackPrFromLam.tofNSigmaPr());

0 commit comments

Comments
 (0)