diff --git a/multiml/agent/basic/sequential.py b/multiml/agent/basic/sequential.py index 840e606..a8fbe65 100644 --- 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