Skip to content

Commit 003a0ea

Browse files
Refactor binsPtMl and cutDirMl to use vectors
1 parent 0067feb commit 003a0ea

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

‎PWGHF/HFC/TableProducer/producerCharmHadronsCharmFemtoDream.cxx‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,10 @@ struct HfProducerCharmHadronsCharmFemtoDream {
7171
struct HfMlConfig : ConfigurableGroup {
7272
std::string prefix;
7373
Configurable<int> mlApplicationMode{"mlApplicationMode", FillMlFromSelector, "0: no ML, 1: selector scores, 2: new BDT after selector"};
74-
Configurable<std::vector<double>> binsPtMl{"binsPtMl", hf_cuts_ml::vecBinsPt, "pT bin limits for new BDT"};
74+
Configurable<std::vector<double>> binsPtMl{"binsPtMl", std::vector<double>{hf_cuts_ml::vecBinsPt}, "pT bin limits for new BDT"};
7575
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::labelsDmesCutScore}, "New BDT cuts per pT bin: background, prompt, nonprompt"};
76-
Configurable<std::vector<int>> cutDirMl{"cutDirMl", hf_cuts_ml::vecCutDir, "Reject scores above (0), below (1), or do not cut (2)"};
77-
Configurable<int> nClassesMl{"nClassesMl", hf_cuts_ml::NCutScores, "Three output classes: background, prompt, nonprompt"};
76+
Configurable<std::vector<int>> cutDirMl{"cutDirMl", std::vector<int>{hf_cuts_ml::vecCutDir}, "Reject scores above (0), below (1), or do not cut (2)"};
77+
Configurable<int> nClassesMl{"nClassesMl", static_cast<int>(hf_cuts_ml::NCutScores), "Three output classes: background, prompt, nonprompt"};
7878
Configurable<std::vector<std::string>> namesInputFeatures{"namesInputFeatures", std::vector<std::string>{}, "Ordered input feature names for new BDT"};
7979
Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{}, "Model files, one per pT bin"};
8080
Configurable<std::vector<std::string>> modelPathsCCDB{"modelPathsCCDB", std::vector<std::string>{}, "CCDB model paths, one per pT bin"};

0 commit comments

Comments
 (0)