From 6b09ade60557328c8feec8ef7bb1a0a5ce7f1549 Mon Sep 17 00:00:00 2001
From: Linlang <30293408+SunsetWolf@users.noreply.github.com>
Date: Thu, 3 Apr 2025 12:01:10 +0800
Subject: [PATCH] fix: fix KeyError direct_exp_gen (#735)

* fix KeyError direct_exp_gen

* fix KeyError direct_exp_gen

* fix fin_factor fetch data error
---
 rdagent/app/qlib_rd_loop/factor_from_report.py | 8 ++++----
 rdagent/scenarios/qlib/docker/Dockerfile       | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/rdagent/app/qlib_rd_loop/factor_from_report.py b/rdagent/app/qlib_rd_loop/factor_from_report.py
index 16aa699e..4917509d 100644
--- a/rdagent/app/qlib_rd_loop/factor_from_report.py
+++ b/rdagent/app/qlib_rd_loop/factor_from_report.py
@@ -116,7 +116,7 @@ class FactorReportLoop(FactorRDLoop, metaclass=LoopMeta):
         self.valid_pdf_file_count = 0
         self.current_loop_hypothesis = None
         self.current_loop_exp = None
-        self.steps = ["propose_hypo_exp", "propose", "exp_gen", "coding", "running", "feedback"]
+        self.steps = ["propose_hypo_exp", "propose", "direct_exp_gen", "coding", "running", "feedback"]
 
     def propose_hypo_exp(self, prev_out: dict[str, Any]):
         with logger.tag("r"):
@@ -144,12 +144,12 @@ class FactorReportLoop(FactorRDLoop, metaclass=LoopMeta):
     def propose(self, prev_out: dict[str, Any]):
         return self.current_loop_hypothesis
 
-    def exp_gen(self, prev_out: dict[str, Any]):
-        return self.current_loop_exp
+    def direct_exp_gen(self, prev_out: dict[str, Any]):
+        return {"propose": self.current_loop_hypothesis, "exp_gen": self.current_loop_exp}
 
     def coding(self, prev_out: dict[str, Any]):
         with logger.tag("d"):  # develop
-            exp = self.coder.develop(prev_out["exp_gen"])
+            exp = self.coder.develop(prev_out["direct_exp_gen"]["exp_gen"])
             logger.log_object(exp.sub_workspace_list, tag="coder result")
         return exp
 
diff --git a/rdagent/scenarios/qlib/docker/Dockerfile b/rdagent/scenarios/qlib/docker/Dockerfile
index df75da5d..2cc727f3 100644
--- a/rdagent/scenarios/qlib/docker/Dockerfile
+++ b/rdagent/scenarios/qlib/docker/Dockerfile
@@ -13,7 +13,7 @@ RUN git clone --depth=100 https://github.com/microsoft/qlib.git
 
 WORKDIR /workspace/qlib
 
-RUN git reset 4d621bff99bb403231e26bb1dddae6e77d2af170 --hard
+RUN git reset e7a1b5ea1ff4412792eeb606a639e0dde422053a --hard
 
 RUN python -m pip install --upgrade cython
 RUN python -m pip install -e .