From b18471a11235e1af0eea9e7b07e4c0d9c091689f Mon Sep 17 00:00:00 2001 From: tomoe Date: Wed, 14 Jun 2023 15:33:36 +0900 Subject: [PATCH] Fix subtask_id bug --- multiml/agent/basic/sequential.py | 2 ++ 1 file changed, 2 insertions(+) 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