@@ -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