From 4017780b05f706fdcbde6c354c5e29de12ff2580 Mon Sep 17 00:00:00 2001 From: Travis Wrightsman Date: Thu, 6 Jun 2024 14:19:40 -0400 Subject: [PATCH] Fix pandas FutureWarning --- compleasm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/compleasm.py b/compleasm.py index 65fec28..f90154c 100755 --- a/compleasm.py +++ b/compleasm.py @@ -1278,7 +1278,7 @@ def Run_busco_mode(self): # remaining genes for gene_id in filtered_species: - mapped_records = grouped_df.get_group(gene_id) + mapped_records = grouped_df.get_group((gene_id,)) mapped_records = mapped_records.sort_values(by=["I+L"], ascending=False) if self.specified_contigs is not None: mapped_records = mapped_records[mapped_records["Contig_id"].isin(self.specified_contigs)] @@ -1541,7 +1541,7 @@ def Run_lite_mode(self): full_table_busco_format_writer.write( "# Busco id\tStatus\tSequence\tGene Start\tGene End\tStrand\tScore\tLength\tOrthoDB url\tDescription\n") for gene_id in all_species: - mapped_records = grouped_df.get_group(gene_id) + mapped_records = grouped_df.get_group((gene_id,)) mapped_records = mapped_records.sort_values(by=["I+L"], ascending=False) if self.specified_contigs is not None: mapped_records = mapped_records[mapped_records["Contig_id"].isin(self.specified_contigs)]