From bdca8fce4ddea8f9c43fca731956920e447169a3 Mon Sep 17 00:00:00 2001 From: WinstonLiyt <104308117+WinstonLiyt@users.noreply.github.com> Date: Sat, 28 Sep 2024 03:55:41 +0800 Subject: [PATCH] fix a bug in code costeer (#373) --- .../coder/factor_coder/CoSTEER/evaluators.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py b/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py index f0df8365..3c545346 100644 --- a/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py +++ b/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py @@ -448,10 +448,16 @@ class FactorValueEvaluator(FactorEvaluator): if gt_implementation is not None and (equal_value_ratio_result > 0.99) or high_correlation_result: decision_from_value_check = True - elif row_result <= 0.99 or output_format_result is False or daily_check_result is False: - decision_from_value_check = False - else: - decision_from_value_check = None + if version == 1: + if row_result <= 0.99 or output_format_result is False or daily_check_result is False: + decision_from_value_check = False + else: + decision_from_value_check = None + elif version == 2: + if output_format_result is False or daily_check_result is False: + decision_from_value_check = False + else: + decision_from_value_check = None return conclusion_str, decision_from_value_check