fix bug.

bigbio · Dec 3, 2024 · 9f54092 · 9f54092
1 parent 6b9a2b1
commit 9f54092
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/quantmsutils/psm/psm_conversion.py b/quantmsutils/psm/psm_conversion.py
@@ -7,6 +7,8 @@
 import pandas as pd
 import pyopenms as oms
 
+from quantmsutils.utils.constants import SCAN, MZ_ARRAY, INTENSITY_ARRAY
+
 _parquet_field = [
     "sequence",
     "protein_accessions",
@@ -113,9 +115,9 @@ def convert_psm(
         )
 
         if isinstance(spectra_df, pd.DataFrame):
-            spectra = spectra_df[spectra_df["scan"] == scan_number]
-            mz_array = spectra["mz"].values
-            intensity_array = spectra["intensity"].values
+            spectra = spectra_df[spectra_df[SCAN] == scan_number]
+            mz_array = spectra[MZ_ARRAY].values
+            intensity_array = spectra[INTENSITY_ARRAY].values
             num_peaks = len(mz_array)
 
         for hit in peptide_id.getHits():