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