@@ -593,7 +593,7 @@ struct TreeCreatorElectronMLDDA {
593593 // float dcaZ = mDcaInfoCov.getZ();
594594
595595 if (pidlabel == static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kElectron )) {
596- if (trackParCov. getP () < max_p_for_downscaling_electron) {
596+ if (track. tpcInnerParam () < max_p_for_downscaling_electron) {
597597 if (dist01 (engine) > downscaling_electron_lowP) {
598598 return ;
599599 }
@@ -603,7 +603,7 @@ struct TreeCreatorElectronMLDDA {
603603 }
604604 }
605605 } else if (pidlabel == static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kPion )) {
606- if (trackParCov. getP () < max_p_for_downscaling_pion) {
606+ if (track. tpcInnerParam () < max_p_for_downscaling_pion) {
607607 if (dist01 (engine) > downscaling_pion_lowP) {
608608 return ;
609609 }
@@ -613,7 +613,7 @@ struct TreeCreatorElectronMLDDA {
613613 }
614614 }
615615 } else if (pidlabel == static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kKaon )) {
616- if (trackParCov. getP () < max_p_for_downscaling_kaon) {
616+ if (track. tpcInnerParam () < max_p_for_downscaling_kaon) {
617617 if (dist01 (engine) > downscaling_kaon_lowP) {
618618 return ;
619619 }
@@ -623,7 +623,7 @@ struct TreeCreatorElectronMLDDA {
623623 }
624624 }
625625 } else if (pidlabel == static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kProton )) {
626- if (trackParCov. getP () < max_p_for_downscaling_proton) {
626+ if (track. tpcInnerParam () < max_p_for_downscaling_proton) {
627627 if (dist01 (engine) > downscaling_proton_lowP) {
628628 return ;
629629 }
0 commit comments