fix: support coeff=None for continuous factors in lfc_shrink()

owkin · Aug 1, 2023 · 61838d1 · 61838d1
1 parent a0ff4a4
commit 61838d1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pydeseq2/ds.py b/pydeseq2/ds.py
@@ -304,7 +304,14 @@ def lfc_shrink(self, coeff: Optional[str] = None) -> None:
             (default: ``None``).
         """
 
-        contrast_level = f"{self.contrast[0]}_{self.contrast[1]}_vs_{self.contrast[2]}"
+        if self.contrast[1] == self.contrast[2] == "":
+            # The factor being tested is continuous
+            contrast_level = self.contrast[0]
+        else:
+            # The factor being tested is categorical
+            contrast_level = (
+                f"{self.contrast[0]}_{self.contrast[1]}_vs_{self.contrast[2]}"
+            )
 
         if coeff is not None:
             if coeff not in self.LFC.columns: