diff --git a/EventFiltering/PWGHF/HFFilter.cxx b/EventFiltering/PWGHF/HFFilter.cxx index 9dfaeb28300..9d42b18de1c 100644 --- a/EventFiltering/PWGHF/HFFilter.cxx +++ b/EventFiltering/PWGHF/HFFilter.cxx @@ -933,7 +933,11 @@ struct HfFilter { // Main struct for HF triggers // TODO: add more feature configurations std::vector inputFeatures{trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ()}; std::vector inputFeaturesD{trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ()}; - for (auto iCharmPart{0}; (iCharmPart < kNCharmParticles - 1) && is3Prong[iCharmPart] && onnxFiles[iCharmPart + 1] != ""; ++iCharmPart) { + for (auto iCharmPart{0}; iCharmPart < kNCharmParticles - 1; ++iCharmPart) { + if (!is3Prong[iCharmPart] || onnxFiles[iCharmPart + 1] == "") { + continue; + } + std::vector inputTensor; if (dataTypeML[iCharmPart + 1] == 1) { inputTensor.push_back(Ort::Experimental::Value::CreateTensor(inputFeatures.data(), inputFeatures.size(), inputShapesML[iCharmPart + 1][0]));