[fix] FutureWarning: Downcasting behavior

- Pandas 3.0 will not auto change type from objects to e.g. numerical use infer_types
cbg-ethz · Jul 3, 2024 · 77a8b9a · 77a8b9a
1 parent 42dd808
commit 77a8b9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lollipop/preprocessors.py b/lollipop/preprocessors.py
@@ -97,7 +97,7 @@ def general_preprocess(
         # self.df_tally = self
         self.df_tally = self.df_tally.replace(
             ["extra", "mut", "shared", "revert", "subset"], 1
-        )
+        ).infer_objects()
 
         # remove uninformative mutations
         variants_columns = list(set(variants_list) & set(self.df_tally.columns))