silences deprecation warning

domonik · Aug 21, 2024 · 5a2cfbb · 5a2cfbb
1 parent bc5ca47
commit 5a2cfbb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/RAPDOR/datastructures.py b/RAPDOR/datastructures.py
@@ -277,7 +277,7 @@ def _set_design_and_array(self):
         self.array = array
         self.internal_design_matrix = tmp
         indices = self.internal_design_matrix.groupby("Treatment", group_keys=True, observed=False).apply(
-            lambda x: list(x.index))
+            lambda x: list(x.index), include_groups=False)
         self.indices = tuple(np.asarray(index) for index in indices)
 
         p = ~np.any(self.array, axis=-1)

diff --git a/RAPDOR/tests/test_datastructures.py b/RAPDOR/tests/test_datastructures.py
@@ -70,11 +70,11 @@ def multi_intensities(intensities):
 
 def drop_replicates(design, rnase_rep, ctrl_rep):
     if rnase_rep > 0:
-        rnase = design[design["Treatment"] == "RNase"].groupby("Replicate").apply(lambda x: list(x.index)).sample(n=rnase_rep).sum()
+        rnase = design[design["Treatment"] == "RNase"].groupby("Replicate").apply(lambda x: list(x.index), include_groups=False).sample(n=rnase_rep).sum()
     else:
         rnase = []
     if ctrl_rep > 0:
-        ctrl = design[design["Treatment"] == "Control"].groupby("Replicate").apply(lambda x: list(x.index)).sample(n=ctrl_rep).sum()
+        ctrl = design[design["Treatment"] == "Control"].groupby("Replicate").apply(lambda x: list(x.index), include_groups=False).sample(n=ctrl_rep).sum()
     else:
         ctrl = []
     rnase = design.loc[rnase]