Update gen.py (#576)

microsoft · Aug 25, 2021 · 84d2cb3 · 84d2cb3
1 parent 509bfcb
commit 84d2cb3
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