@@ -594,7 +594,7 @@ struct TreeCreatorElectronMLDDA {
594594 // float dcaZ = mDcaInfoCov.getZ();
595595
596596 if (pidlabel == static_cast<uint8_t>(o2::aod::pwgem::dilepton::ml::PID_Label::kElectron)) {
597- if (trackParCov.getP () < max_p_for_downscaling_electron) {
597+ if (track.tpcInnerParam () < max_p_for_downscaling_electron) {
598598 if (dist01(engine) > downscaling_electron_lowP) {
599599 return;
600600 }
@@ -604,7 +604,7 @@ struct TreeCreatorElectronMLDDA {
604604 }
605605 }
606606 } else if (pidlabel == static_cast<uint8_t>(o2::aod::pwgem::dilepton::ml::PID_Label::kPion)) {
607- if (trackParCov.getP () < max_p_for_downscaling_pion) {
607+ if (track.tpcInnerParam () < max_p_for_downscaling_pion) {
608608 if (dist01(engine) > downscaling_pion_lowP) {
609609 return;
610610 }
@@ -614,7 +614,7 @@ struct TreeCreatorElectronMLDDA {
614614 }
615615 }
616616 } else if (pidlabel == static_cast<uint8_t>(o2::aod::pwgem::dilepton::ml::PID_Label::kKaon)) {
617- if (trackParCov.getP () < max_p_for_downscaling_kaon) {
617+ if (track.tpcInnerParam () < max_p_for_downscaling_kaon) {
618618 if (dist01(engine) > downscaling_kaon_lowP) {
619619 return;
620620 }
@@ -624,7 +624,7 @@ struct TreeCreatorElectronMLDDA {
624624 }
625625 }
626626 } else if (pidlabel == static_cast<uint8_t>(o2::aod::pwgem::dilepton::ml::PID_Label::kProton)) {
627- if (trackParCov.getP () < max_p_for_downscaling_proton) {
627+ if (track.tpcInnerParam () < max_p_for_downscaling_proton) {
628628 if (dist01(engine) > downscaling_proton_lowP) {
629629 return;
630630 }
0 commit comments