From 890661735da4cbcb9017e53ab5e3884d9b64588a Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Thu, 13 Mar 2025 14:55:13 +0800
Subject: [PATCH] fix: save only one mle_score pkl for a running exp (#675)

* save only one mle_score pkl for a running exp

* fix_CI
---
 rdagent/log/mle_summary.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/rdagent/log/mle_summary.py b/rdagent/log/mle_summary.py
index 129d43ac..8a8a3c38 100644
--- a/rdagent/log/mle_summary.py
+++ b/rdagent/log/mle_summary.py
@@ -46,7 +46,9 @@ def save_grade_info(log_trace_path: Path):
                     entry=f"mlebench grade-sample submission.csv {competition} --data-dir /mle/data | tee mle_score.txt",
                 )
                 msg.content.experiment_workspace.execute(env=de, entry="chmod 777 mle_score.txt")
-                trace_storage.log(mle_score_str, name=f"{msg.tag}.mle_score.pid", save_type="pkl")
+                trace_storage.log(
+                    mle_score_str, name=f"{msg.tag}.mle_score.pid", save_type="pkl", timestamp=msg.timestamp
+                )
 
 
 def is_valid_session(p: Path) -> bool: