From 249d6f9810e460014132bbdcfd84fc8c2efb191a Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Thu, 29 May 2025 16:40:57 +0800
Subject: [PATCH] fix loop_num calc logic (#911)

---
 rdagent/log/mle_summary.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/rdagent/log/mle_summary.py b/rdagent/log/mle_summary.py
index 12b185be..2f75e3d0 100644
--- a/rdagent/log/mle_summary.py
+++ b/rdagent/log/mle_summary.py
@@ -1,6 +1,4 @@
-import json
 import pickle
-import re
 from collections import defaultdict
 from pathlib import Path
 
@@ -115,12 +113,12 @@ def summarize_folder(log_folder: Path, hours: int | None = None):
             loop_id, fn = extract_loopid_func_name(msg.tag)
             if loop_id:
                 loop_id = int(loop_id)
+                loop_num = max(loop_id + 1, loop_num)
             if hours and loop_id == stop_li and fn == stop_fn:
                 break
             if msg.tag and "llm" not in msg.tag and "session" not in msg.tag:
                 if "competition" in msg.tag:
                     stat[log_trace_path.name]["competition"] = msg.content
-                    start_time = msg.timestamp
 
                     # get threshold scores
                     workflowexp = FBWorkspace()
@@ -136,9 +134,6 @@ def summarize_folder(log_folder: Path, hours: int | None = None):
                             gold_threshold = grade_output["gold_threshold"]
                             median_threshold = grade_output["median_threshold"]
 
-                if "direct_exp_gen" in msg.tag and isinstance(msg.content, DSExperiment):
-                    loop_num += 1
-
                 if "running" in msg.tag:
                     if isinstance(msg.content, DSExperiment):
                         if msg.content.result is not None: