From e75706c8ecb880786ae87cd7ef239259231f8d28 Mon Sep 17 00:00:00 2001 From: XianBW <36835909+XianBW@users.noreply.github.com> Date: Thu, 12 Sep 2024 18:04:21 +0800 Subject: [PATCH] fix rdagent factor bug when FactorTask init (#261) --- rdagent/scenarios/qlib/proposal/factor_proposal.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/rdagent/scenarios/qlib/proposal/factor_proposal.py b/rdagent/scenarios/qlib/proposal/factor_proposal.py index daa43baf..ef58fc98 100644 --- a/rdagent/scenarios/qlib/proposal/factor_proposal.py +++ b/rdagent/scenarios/qlib/proposal/factor_proposal.py @@ -84,7 +84,14 @@ class QlibFactorHypothesis2Experiment(FactorHypothesis2Experiment): description = response_dict[factor_name]["description"] formulation = response_dict[factor_name]["formulation"] variables = response_dict[factor_name]["variables"] - tasks.append(FactorTask(factor_name, description, formulation, variables)) + tasks.append( + FactorTask( + factor_name=factor_name, + factor_description=description, + factor_formulation=formulation, + variables=variables, + ) + ) exp = QlibFactorExperiment(tasks) exp.based_experiments = [QlibFactorExperiment(sub_tasks=[])] + [t[1] for t in trace.hist if t[2]]