Fix subtask_id bug

UTokyo-ICEPP · Jun 14, 2023 · b18471a · b18471a
1 parent 8f74876
commit b18471a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/multiml/agent/basic/sequential.py b/multiml/agent/basic/sequential.py
@@ -113,6 +113,7 @@ def execute_pipeline(self, subtasktuples, counter, trial=None):
             subtask_env = subtasktuple.env
             subtask_hps = copy.deepcopy(subtasktuple.hps)
 
+            subtask_env.subtask_id = subtask_id
             subtask_env.saver = self._saver
             subtask_env.storegate = self._storegate
             subtask_env.job_id = counter
@@ -149,6 +150,7 @@ def execute_differentiable(self, subtasktuples, counter, trial=None):
                 subtask_env = subtasktuple.env
                 subtask_hps = copy.deepcopy(subtasktuple.hps)
 
+                subtask_env.subtask_id = subtask_id
                 subtask_env.saver = self._saver
                 subtask_env.storegate = self._storegate
                 subtask_env.job_id = counter