Update gen.py (microsoft#576)

qianyun210603 · Aug 25, 2021 · 5d95d54 · 5d95d54
1 parent b7bcd44
commit 5d95d54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qlib/workflow/task/gen.py b/qlib/workflow/task/gen.py
@@ -162,8 +162,7 @@ def gen_following_tasks(self, task: dict, test_end: pd.Timestamp) -> List[dict]:
         List[dict]:
             the following tasks of `task`(`task` itself is excluded)
         """
-        t = copy.deepcopy(task)
-        prev_seg = t["dataset"]["kwargs"]["segments"]
+        prev_seg = task["dataset"]["kwargs"]["segments"]
         while True:
             segments = {}
             try:
@@ -184,6 +183,7 @@ def gen_following_tasks(self, task: dict, test_end: pd.Timestamp) -> List[dict]:
                 break
 
             prev_seg = segments
+            t = copy.deepcopy(task)  # deepcopy is necessary to avoid modify task inplace
             self._update_task_segs(t, segments)
             yield t