diff --git a/cayman/annotate/crazy_annotator.py b/cayman/annotate/crazy_annotator.py index 19aa05f..b100fc8 100644 --- 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):