From 430ba3a4ffde0cf467555ea3cc7008dc22aa01b0 Mon Sep 17 00:00:00 2001
From: fangfangssj <1135470306@qq.com>
Date: Tue, 3 Feb 2026 03:21:28 +0000
Subject: [PATCH] fix_es

---
 graph_net_visual/plot_ESt.py | 42 +++++++++---------------------------
 1 file changed, 10 insertions(+), 32 deletions(-)

diff --git a/graph_net_visual/plot_ESt.py b/graph_net_visual/plot_ESt.py
index f44a57f4f6..5c4988c149 100644
--- a/graph_net_visual/plot_ESt.py
+++ b/graph_net_visual/plot_ESt.py
@@ -348,38 +348,16 @@ def main(args):
                         np.where(agg_x_vals == 0)[0][0] if 0 in agg_x_vals else None
                     )
 
-                    if agg_zero_index is not None:
-                        ax.plot(
-                            agg_x_vals[: agg_zero_index + 1],
-                            agg_y_vals[: agg_zero_index + 1],
-                            "s--",
-                            color=color,
-                            label=f"{folder_name} (aggregated)",
-                            linewidth=2,
-                            markersize=6,
-                            alpha=0.7,
-                        )
-                        ax.plot(
-                            agg_x_vals[agg_zero_index:],
-                            agg_y_vals[agg_zero_index:],
-                            "s--",
-                            color=color,
-                            linewidth=2,
-                            markersize=6,
-                            drawstyle="steps-post",
-                            alpha=0.7,
-                        )
-                    else:
-                        ax.plot(
-                            agg_x_vals,
-                            agg_y_vals,
-                            "s--",
-                            color=color,
-                            label=f"{folder_name} (aggregated)",
-                            linewidth=2,
-                            markersize=6,
-                            alpha=0.7,
-                        )
+                    ax.plot(
+                        agg_x_vals[agg_zero_index:],
+                        agg_y_vals[agg_zero_index:],
+                        "s--",
+                        color=color,
+                        linewidth=2,
+                        markersize=6,
+                        drawstyle="steps-post",
+                        alpha=0.7,
+                    )
 
             # Update x-axis range if needed
             if all_x_coords: