Skip to content

Commit 9721c8a

Browse files
author
Mattia Faggin
committed
Fix getInputFeature after rebase.
1 parent 6517e3f commit 9721c8a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

PWGHF/HFC/TableProducer/femtoDreamProducer.cxx

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -145,8 +145,8 @@ struct HfFemtoDreamProducer {
145145

146146
float magField;
147147
int runNumber;
148-
using CandidateLc = soa::Join<aod::HfCand3Prong, aod::HfSelLc>;
149-
using CandidateLcMc = soa::Join<aod::HfCand3Prong, aod::HfSelLc, aod::HfCand3ProngMcRec>;
148+
using CandidateLc = soa::Join<aod::HfCand3ProngWPid, aod::HfSelLc>;
149+
using CandidateLcMc = soa::Join<aod::HfCand3ProngWPid, aod::HfSelLc, aod::HfCand3ProngMcRec>;
150150

151151
using FemtoFullCollision = soa::Join<aod::Collisions, aod::EvSels, aod::Mults, aod::CentFT0Ms>::iterator;
152152
using FemtoFullCollisionMc = soa::Join<aod::Collisions, aod::EvSels, aod::Mults, aod::CentFT0Ms, aod::McCollisionLabels>::iterator;
@@ -457,11 +457,11 @@ struct HfFemtoDreamProducer {
457457
isSelectedMlLcToPKPi = false;
458458
isSelectedMlLcToPiKP = false;
459459
if (candidate.mlProbLcToPKPi().size() > 0) {
460-
std::vector<float> inputFeaturesLcToPKPi = hfMlResponse.getInputFeatures(candidate, trackPos1, trackNeg, trackPos2, true);
460+
std::vector<float> inputFeaturesLcToPKPi = hfMlResponse.getInputFeatures(candidate, true);
461461
isSelectedMlLcToPKPi = hfMlResponse.isSelectedMl(inputFeaturesLcToPKPi, candidate.pt(), outputMlPKPi);
462462
}
463463
if (candidate.mlProbLcToPiKP().size() > 0) {
464-
std::vector<float> inputFeaturesLcToPiKP = hfMlResponse.getInputFeatures(candidate, trackPos1, trackNeg, trackPos2, false);
464+
std::vector<float> inputFeaturesLcToPiKP = hfMlResponse.getInputFeatures(candidate, false);
465465
isSelectedMlLcToPiKP = hfMlResponse.isSelectedMl(inputFeaturesLcToPiKP, candidate.pt(), outputMlPKPi);
466466
}
467467
if (!isSelectedMlLcToPKPi && !isSelectedMlLcToPiKP)

0 commit comments

Comments
 (0)