Fixing pandas FutureWarning zellerlab#1

karchern · Oct 9, 2024 · 8f601d4 · 8f601d4
1 parent 3a0c5a9
commit 8f601d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cayman/annotate/crazy_annotator.py b/cayman/annotate/crazy_annotator.py
@@ -105,7 +105,7 @@ def curate_annotations(self, precomputed_hmm_cutoffs):
         import pandas as pd
         median_cutoffs = pd.read_csv(precomputed_hmm_cutoffs)
         median_cutoffs['familyType'] = [sub("\d", "", x.replace("_", "")) for x in median_cutoffs['family']]
-        median_cutoffs = median_cutoffs.groupby('familyType').median()['cutoff']
+        median_cutoffs = median_cutoffs.groupby('familyType').median(numeric_only=True)['cutoff']
         cutoffs_all = pd.read_csv(precomputed_hmm_cutoffs)
         family_fold_results_filtered = []
         for index, family_fold_results in enumerate(self.annotations_by_family_and_fold):