Skip to content

Commit 07b96e7

Browse files
add tracking to spectrum file reading
1 parent 8e793a0 commit 07b96e7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ms2rescore/parse_spectra.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
import numpy as np
99
from ms2rescore_rs import Precursor, get_ms2_spectra, MS2Spectrum
10+
from rich.progress import track
1011

1112
from psm_utils import PSMList
1213

@@ -208,7 +209,7 @@ def _add_precursor_values(
208209
if spectrum_id_pattern is None:
209210
spectrum_id_pattern = r"^(.*)$" # Match entire identifier if no pattern provided
210211

211-
for run_name in set(psm_list["run"]):
212+
for run_name in track(set(psm_list["run"])):
212213
run_mask = psm_list["run"] == run_name
213214
psm_list_run = psm_list[run_mask]
214215
spectrum_file = infer_spectrum_path(spectrum_path, run_name)

0 commit comments

Comments
 (0)