Skip to content

Commit a178b6d

Browse files
authored
[Tutorial] Fix int8_t configurables (AliceO2Group#8524)
1 parent d5bae59 commit a178b6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Tutorials/ML/applyMlSelection.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ struct applyMlSelection {
3939
Configurable<std::vector<double>> binsPtMl{"binsPtMl", std::vector<double>{1., 36.}, "pT bin limits for ML application"};
4040
Configurable<std::vector<int>> cutDirMl{"cutDirMl", std::vector<int>{cuts_ml::CutSmaller, cuts_ml::CutNot, cuts_ml::CutNot}, "Whether to reject score values greater or smaller than the threshold"};
4141
Configurable<LabeledArray<double>> cutsMl{"cutsMl", {defaultCutsMl[0], 1, 3, {"pT bin 0"}, {"score prompt", "score non-prompt", "score bkg"}}, "ML selections per pT bin"};
42-
Configurable<int8_t> nClassesMl{"nClassesMl", (int8_t)3, "Number of classes in ML model"};
42+
Configurable<int> nClassesMl{"nClassesMl", 3, "Number of classes in ML model"};
4343
// Model file names
4444
Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"model_onnx.onnx"}, "ONNX file names for each pT bin (if not from CCDB full path)"};
4545
// Bonus: CCDB configuration (needed for ML application on the GRID)

0 commit comments

Comments
 (0)