Skip to content

Commit f65bbc0

Browse files
authored
small bug fix in feedback (microsoft#771)
1 parent 59a6a7b commit f65bbc0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

rdagent/scenarios/data_science/dev/feedback.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,15 +70,15 @@ def generate_feedback(self, exp: DSExperiment, trace: DSTrace) -> ExperimentFeed
7070
decision=True if self.scen.metric_direction else False,
7171
)
7272
elif cur_score < sota_score:
73-
return ExperimentFeedback(
73+
return HypothesisFeedback(
7474
observations="The current score smaller than the SOTA score.",
7575
hypothesis_evaluation="The current score is smaller than the SOTA score.",
7676
new_hypothesis="No new hypothesis provided",
7777
reason="The current score is smaller than the SOTA score.",
7878
decision=False if self.scen.metric_direction else True,
7979
)
8080
else:
81-
return ExperimentFeedback(
81+
return HypothesisFeedback(
8282
observations="The current score equals to the SOTA score.",
8383
hypothesis_evaluation="The current score equals to the SOTA score.",
8484
new_hypothesis="No new hypothesis provided",

0 commit comments

Comments
 (0)