checkpointing

nf-core · Dec 16, 2024 · 9278626 · 9278626
1 parent 9bcf376
commit 9278626
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/train_and_predict_final.py b/bin/train_and_predict_final.py
@@ -209,7 +209,7 @@ def compute_cross(
             predictions_path,
             f"predictions_{args.split_id}.csv",
         )
-        test_set.save(prediction_dataset)
+        test_set.to_csv(prediction_dataset)
         for ds in args.cross_study_datasets:
             if ds == "NONE.csv":
                 continue