@@ -670,7 +670,7 @@ def getTransitionGroupFeaturesDf(self, run_basename_wo_ext: str, fullpeptidename
670670 return features [self .columns ]
671671
672672 def getTopTransitionGroupFeatureDf (self , run_basename_wo_ext : str , fullpeptidename : str , charge : int ) -> pd .DataFrame :
673- columns = ['filename' , 'leftBoundary' , 'rightBoundary' , 'areaIntensity' , 'qvalue' , 'consensusApex' , 'consensusApexIntensity' , 'sequence' , 'precursor_charge' , 'software' ]
673+ columns = self . columns # ['filename', 'leftBoundary', 'rightBoundary', 'areaIntensity', 'qvalue', 'consensusApex', 'consensusApexIntensity', 'sequence', 'precursor_charge', 'software']
674674 run_id = self ._runIDFromRunName (run_basename_wo_ext )
675675 precursor_id = self .getPrecursorIDFromPeptideAndCharge (fullpeptidename , charge )
676676
@@ -681,7 +681,7 @@ def getTopTransitionGroupFeatureDf(self, run_basename_wo_ext: str, fullpeptidena
681681
682682 features ['sequence' ] = fullpeptidename
683683 features ['precursor_charge' ] = charge
684- return features [[ 'leftBoundary' , 'rightBoundary' , 'areaIntensity' , 'qvalue' , 'consensusApex' , 'consensusApexIntensity' , 'sequence' , 'precursor_charge' , 'software' ] ]
684+ return features [self . columns ]
685685
686686 def getTransitionGroupFeatures (self , run_basename_wo_ext : str , fullpeptidename : str , charge : int ) -> List [TransitionGroupFeature ]:
687687 run_id = self ._runIDFromRunName (run_basename_wo_ext )
0 commit comments