From 6d123d75091497fabdba5ea49dc74b1cc85d2fb1 Mon Sep 17 00:00:00 2001 From: dvanaken Date: Fri, 3 Apr 2020 02:30:04 -0400 Subject: [PATCH] Bug fix: pass single result object to save_execution_time() --- server/website/website/tasks/async_tasks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/website/website/tasks/async_tasks.py b/server/website/website/tasks/async_tasks.py index 7d08486..7c87171 100644 --- a/server/website/website/tasks/async_tasks.py +++ b/server/website/website/tasks/async_tasks.py @@ -624,7 +624,7 @@ def train_ddpg(train_ddpg_input): session.ddpg_actor_model, session.ddpg_critic_model = ddpg.get_model() session.ddpg_reply_memory = ddpg.replay_memory.get() session.save() - exec_time = save_execution_time(start_ts, "train_ddpg", result) + exec_time = save_execution_time(start_ts, "train_ddpg", result.first()) LOG.debug("\n%s: Result = %s\n", task_name, _task_result_tostring(target_data)) LOG.info('%s: Done training ddpg (%.1f seconds).', task_name, exec_time) return target_data, algorithm