fix: save only one mle_score pkl for a running exp (#675)

XianBW · WinstonLiyt · commit a3bce93d4d4f · 2025-03-14T09:32:52.000Z
* save only one mle_score pkl for a running exp

* fix_CI
diff --git 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:

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,9 @@ def save_grade_info(log_trace_path: Path):`
`46`	`46`	`entry=f"mlebench grade-sample submission.csv {competition} --data-dir /mle/data \| tee mle_score.txt",`
`47`	`47`	`)`
`48`	`48`	`msg.content.experiment_workspace.execute(env=de, entry="chmod 777 mle_score.txt")`
`49`		`- trace_storage.log(mle_score_str, name=f"{msg.tag}.mle_score.pid", save_type="pkl")`
	`49`	`+ trace_storage.log(`
	`50`	`+ mle_score_str, name=f"{msg.tag}.mle_score.pid", save_type="pkl", timestamp=msg.timestamp`
	`51`	`+ )`
`50`	`52`
`51`	`53`
`52`	`54`	`def is_valid_session(p: Path) -> bool:`