Skip to content

Commit 7200682

Browse files
authored
fix: fix KeyError direct_exp_gen (#735)
* fix KeyError direct_exp_gen * fix KeyError direct_exp_gen * fix fin_factor fetch data error
1 parent ab9c7b9 commit 7200682

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

rdagent/app/qlib_rd_loop/factor_from_report.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def __init__(self, report_folder: str = None):
116116
self.valid_pdf_file_count = 0
117117
self.current_loop_hypothesis = None
118118
self.current_loop_exp = None
119-
self.steps = ["propose_hypo_exp", "propose", "exp_gen", "coding", "running", "feedback"]
119+
self.steps = ["propose_hypo_exp", "propose", "direct_exp_gen", "coding", "running", "feedback"]
120120

121121
def propose_hypo_exp(self, prev_out: dict[str, Any]):
122122
with logger.tag("r"):
@@ -144,12 +144,12 @@ def propose_hypo_exp(self, prev_out: dict[str, Any]):
144144
def propose(self, prev_out: dict[str, Any]):
145145
return self.current_loop_hypothesis
146146

147-
def exp_gen(self, prev_out: dict[str, Any]):
148-
return self.current_loop_exp
147+
def direct_exp_gen(self, prev_out: dict[str, Any]):
148+
return {"propose": self.current_loop_hypothesis, "exp_gen": self.current_loop_exp}
149149

150150
def coding(self, prev_out: dict[str, Any]):
151151
with logger.tag("d"): # develop
152-
exp = self.coder.develop(prev_out["exp_gen"])
152+
exp = self.coder.develop(prev_out["direct_exp_gen"]["exp_gen"])
153153
logger.log_object(exp.sub_workspace_list, tag="coder result")
154154
return exp
155155

rdagent/scenarios/qlib/docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ RUN git clone --depth=100 https://github.com/microsoft/qlib.git
1313

1414
WORKDIR /workspace/qlib
1515

16-
RUN git reset 4d621bff99bb403231e26bb1dddae6e77d2af170 --hard
16+
RUN git reset e7a1b5ea1ff4412792eeb606a639e0dde422053a --hard
1717

1818
RUN python -m pip install --upgrade cython
1919
RUN python -m pip install -e .

0 commit comments

Comments
 (0)