From 2940dc42b4e2d339b2b309c42fbccda53da04f1a Mon Sep 17 00:00:00 2001 From: Himanshu Sharma Date: Sat, 21 Mar 2026 16:24:16 +0100 Subject: [PATCH 1/2] Add check for ML scores array size --- PWGHF/D2H/Tasks/taskD0.cxx | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/PWGHF/D2H/Tasks/taskD0.cxx b/PWGHF/D2H/Tasks/taskD0.cxx index 64d228b45d7..9eb533c5fa7 100644 --- a/PWGHF/D2H/Tasks/taskD0.cxx +++ b/PWGHF/D2H/Tasks/taskD0.cxx @@ -667,9 +667,12 @@ struct HfTaskD0 { valuesToFill.push_back(static_cast(mass)); valuesToFill.push_back(static_cast(ptCandidate)); if constexpr (FillMl) { - valuesToFill.push_back(candidate.mlProbD0()[0]); - valuesToFill.push_back(candidate.mlProbD0()[1]); - valuesToFill.push_back(candidate.mlProbD0()[2]); + auto const& mlScores = candidate.mlProbD0(); + if (mlScores.size() == 3) { + valuesToFill.push_back(mlScores[0]); + valuesToFill.push_back(mlScores[1]); + valuesToFill.push_back(mlScores[2]); + } } valuesToFill.push_back(static_cast(HfHelper::yD0(candidate))); valuesToFill.push_back(static_cast(d0Type)); From 5cf83f4859254384564c34fd94d86b039ad86c1a Mon Sep 17 00:00:00 2001 From: ALICE Action Bot Date: Sat, 21 Mar 2026 15:26:02 +0000 Subject: [PATCH 2/2] Please consider the following formatting changes --- PWGHF/D2H/Tasks/taskD0.cxx | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/PWGHF/D2H/Tasks/taskD0.cxx b/PWGHF/D2H/Tasks/taskD0.cxx index 9eb533c5fa7..e879ba57c9a 100644 --- a/PWGHF/D2H/Tasks/taskD0.cxx +++ b/PWGHF/D2H/Tasks/taskD0.cxx @@ -669,9 +669,9 @@ struct HfTaskD0 { if constexpr (FillMl) { auto const& mlScores = candidate.mlProbD0(); if (mlScores.size() == 3) { - valuesToFill.push_back(mlScores[0]); - valuesToFill.push_back(mlScores[1]); - valuesToFill.push_back(mlScores[2]); + valuesToFill.push_back(mlScores[0]); + valuesToFill.push_back(mlScores[1]); + valuesToFill.push_back(mlScores[2]); } } valuesToFill.push_back(static_cast(HfHelper::yD0(candidate)));