From a16c79e45bc7d0e9f2f9efae3704238f10d2b236 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Tue, 8 Apr 2025 08:17:47 -0600 Subject: [PATCH] small bug fix in feedback (#771) --- rdagent/scenarios/data_science/dev/feedback.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rdagent/scenarios/data_science/dev/feedback.py b/rdagent/scenarios/data_science/dev/feedback.py index 238d2e23..98645deb 100644 --- a/rdagent/scenarios/data_science/dev/feedback.py +++ b/rdagent/scenarios/data_science/dev/feedback.py @@ -70,7 +70,7 @@ class DSExperiment2Feedback(Experiment2Feedback): decision=True if self.scen.metric_direction else False, ) elif cur_score < sota_score: - return ExperimentFeedback( + return HypothesisFeedback( observations="The current score smaller than the SOTA score.", hypothesis_evaluation="The current score is smaller than the SOTA score.", new_hypothesis="No new hypothesis provided", @@ -78,7 +78,7 @@ class DSExperiment2Feedback(Experiment2Feedback): decision=False if self.scen.metric_direction else True, ) else: - return ExperimentFeedback( + return HypothesisFeedback( observations="The current score equals to the SOTA score.", hypothesis_evaluation="The current score equals to the SOTA score.", new_hypothesis="No new hypothesis provided",