From 4b3739c5e8ab92310dedbc0a2b938f62dbfc63be Mon Sep 17 00:00:00 2001
From: Xisen Wang <118058822+Xisen-Wang@users.noreply.github.com>
Date: Fri, 5 Jul 2024 18:14:48 +0800
Subject: [PATCH] Update proposal.py

Updated the summarize function to include more inputs
---
 rdagent/core/proposal.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/rdagent/core/proposal.py b/rdagent/core/proposal.py
index 144f34a7..8143b15a 100644
--- a/rdagent/core/proposal.py
+++ b/rdagent/core/proposal.py
@@ -88,11 +88,10 @@ class Hypothesis2Experiment(ABC, Generic[ASpecificExp]):
 
 # Boolean, Reason, Confidence, etc.
 
-
 class Experiment2Feedback:
     """ "Generated(summarize) feedback from **Executed** Implementation"""
 
-    def summarize(self, ti: Experiment) -> HypothesisFeedback:
+    def summarize(self, ti: Experiment, hypothesis: Hypothesis, trace: Trace) -> HypothesisFeedback:
         """
         The `ti` should be executed and the results should be included.
         For example: `mlflow` of Qlib will be included.