From 88a1b08b8c2124fe2ef8b2b87640876778faf649 Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Mon, 1 Sep 2025 14:46:51 +0800
Subject: [PATCH] for ui, a simple fix (#1215)

---
 rdagent/scenarios/data_science/proposal/exp_gen/merge.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/rdagent/scenarios/data_science/proposal/exp_gen/merge.py b/rdagent/scenarios/data_science/proposal/exp_gen/merge.py
index 29eb54d7..906810e0 100644
--- a/rdagent/scenarios/data_science/proposal/exp_gen/merge.py
+++ b/rdagent/scenarios/data_science/proposal/exp_gen/merge.py
@@ -18,6 +18,10 @@ from rdagent.scenarios.data_science.proposal.exp_gen.proposal import DSProposalV
 from rdagent.utils.agent.tpl import T
 from rdagent.utils.workflow import wait_retry
 
+from .proposal import (
+    HypothesisComponent,  # FIXME: for statistic of other branches after running, remove this later
+)
+
 
 class MergeExpGen(ExpGen):
     def gen(