Fix coverage test

Author: z3rone

xifdr/utils/column_preparation.py

@@ -126,6 +126,22 @@ def prepare_columns(df, decoy_adjunct:str = 'REV_'):
             coverage_p2 = pl.lit(0.5),
         )
 
+
+    # Fill in infinite scores
+    max_score = df.filter(pl.col('score') < np.inf)['score'].max()
+    min_score = df.filter(pl.col('score') > -np.inf)['score'].min()
+    inf_margin = (max_score-min_score)*0.1
+    df = df.with_columns(pl.col('score') - min_score + inf_margin)
+    df = df.with_columns(
+        score=pl.when(pl.col('score') == np.inf).then(
+            pl.lit(max_score) + 2*pl.lit(inf_margin)
+        ).when(pl.col('score') == -np.inf).then(
+            pl.lit(0)
+        ).otherwise(
+            pl.col('score')
+        )
+    )
+
     coverage_p1_prop = pl.col('coverage_p1') / (pl.col('coverage_p1') + pl.col('coverage_p2'))
     coverage_p2_prop = pl.col('coverage_p2') / (pl.col('coverage_p1') + pl.col('coverage_p2'))
     df = df.with_columns(
@@ -204,21 +220,6 @@ def prepare_columns(df, decoy_adjunct:str = 'REV_'):
         DD=(pl.col('decoy_class')=='DD'),
     )
 
-    # Fill in infinite scores
-    max_score = df.filter(pl.col('score') < np.inf)['score'].max()
-    min_score = df.filter(pl.col('score') > -np.inf)['score'].min()
-    inf_margin = (max_score-min_score)*0.1
-    df = df.with_columns(pl.col('score') - min_score + inf_margin)
-    df = df.with_columns(
-        score=pl.when(pl.col('score') == np.inf).then(
-            pl.lit(max_score) + 2*pl.lit(inf_margin)
-        ).when(pl.col('score') == -np.inf).then(
-            pl.lit(0)
-        ).otherwise(
-            pl.col('score')
-        )
-    )
-
     df = df.with_columns(
         pl.col(list_cols_1).replace([], None),
         pl.col(list_cols_2).replace([], None),