fix undeclared var

hgb-bin-proteomics · Jan 24, 2024 · 3972e36 · 3972e36
1 parent 15a6b9a
commit 3972e36
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/msannika_fdr.py b/msannika_fdr.py
@@ -148,7 +148,8 @@ def get_cutoff(data: pd.DataFrame, fdr: float) -> float:
         labels = data["Class_label"].to_numpy()
         labels_sorted = labels[data["Combined Score"].to_numpy().argsort()]
 
-        for i, score in enumerate(sorted(data["Combined Score"].tolist())):
+        scores = sorted(data["Combined Score"].tolist())
+        for i, score in enumerate(scores):
             if labels_sorted[i:].sum() / (labels_sorted[i:].shape[0] - labels_sorted[i:].sum()) < fdr:
                 return score
 
@@ -181,7 +182,8 @@ def get_cutoff(data: pd.DataFrame, fdr: float) -> float:
         labels = data["Class_label"].to_numpy()
         labels_sorted = labels[data["Best CSM Score"].to_numpy().argsort()]
 
-        for i, score in enumerate(sorted(data["Best CSM Score"].tolist())):
+        scores = sorted(data["Best CSM Score"].tolist())
+        for i, score in enumerate(scores):
             if labels_sorted[i:].sum() / (labels_sorted[i:].shape[0] - labels_sorted[i:].sum()) < fdr:
                 return score