Try inserting mol_ids alternatively

BioinfoMachineLearning · Jan 18, 2025 · 6715388 · 6715388
1 parent b049aff
commit 6715388
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/posebench/analysis/inference_analysis.py b/posebench/analysis/inference_analysis.py
@@ -935,7 +935,7 @@ def main(cfg: DictConfig):
             buster = PoseBusters(config="redock", top_n=None)
             buster.config["loading"]["mol_true"]["load_all"] = False
             bust_results = buster.bust_table(mol_table, full_report=cfg.full_report)
-            bust_results.loc[:, "mol_id"] = mol_ids
+            bust_results["mol_id"] = mol_ids.values
 
             bust_results.to_csv(bust_results_filepath, index=False)
             logger.info(