Correct contigs data bug.

koszullab · Apr 5, 2023 · e84cdcd · e84cdcd
1 parent bcbcd1d
commit e84cdcd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metator/validation.py b/metator/validation.py
@@ -64,12 +64,12 @@ def correct_final_bin(contigs_data, final_fasta_dir, bin_summary):
     pandas.core.frame.DataFrame
         Dataframe with the contigs informations.
     """
-    contigs_data.set_index("Name", inplace=True)
+    contigs_data = contigs_data.set_index("Name")
     for bin_name in bin_summary:
         fasta_file = join(final_fasta_dir, f"{bin_name}.fa")
         fasta = pyfastx.Fasta(fasta_file)
         for seq in fasta:
-            contigs_data.loc[seq.name]["Final_bin"] = bin_name
+            contigs_data.loc[seq.name, "Final_bin"] = bin_name
     return contigs_data