Skip to content

Commit daf6f0c

Browse files
authored
Fix BDT application for 3-prong decays (#1104)
1 parent a7ad73f commit daf6f0c

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

EventFiltering/PWGHF/HFFilter.cxx

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -933,7 +933,11 @@ struct HfFilter { // Main struct for HF triggers
933933
// TODO: add more feature configurations
934934
std::vector<float> inputFeatures{trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ()};
935935
std::vector<double> inputFeaturesD{trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ()};
936-
for (auto iCharmPart{0}; (iCharmPart < kNCharmParticles - 1) && is3Prong[iCharmPart] && onnxFiles[iCharmPart + 1] != ""; ++iCharmPart) {
936+
for (auto iCharmPart{0}; iCharmPart < kNCharmParticles - 1; ++iCharmPart) {
937+
if (!is3Prong[iCharmPart] || onnxFiles[iCharmPart + 1] == "") {
938+
continue;
939+
}
940+
937941
std::vector<Ort::Value> inputTensor;
938942
if (dataTypeML[iCharmPart + 1] == 1) {
939943
inputTensor.push_back(Ort::Experimental::Value::CreateTensor<float>(inputFeatures.data(), inputFeatures.size(), inputShapesML[iCharmPart + 1][0]));

0 commit comments

Comments
 (0)