From da8c46e4b3667654073492d8f2b56511a7bc1549 Mon Sep 17 00:00:00 2001 From: leahh Date: Wed, 11 Sep 2024 08:00:47 -0600 Subject: [PATCH] added task_state --- beeflow/tests/mocks.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/beeflow/tests/mocks.py b/beeflow/tests/mocks.py index b5d11c992..1d7cada86 100644 --- a/beeflow/tests/mocks.py +++ b/beeflow/tests/mocks.py @@ -166,10 +166,10 @@ def reset_workflow(self, old_id, new_id): #noqa not using parameter in mock self.task_metadata[task_id] = {} self.task_states[task_id] = 'WAITING' - def load_task(self, task): + def load_task(self, task, task_state): """Load a task into a workflow in the graph database.""" self.tasks[task.id] = task - self.task_states[task.id] = 'WAITING' + self.task_states[task.id] = task_state self.task_metadata[task.id] = {} self.inputs[task.id] = {} self.outputs[task.id] = {} @@ -186,7 +186,8 @@ def initialize_ready_tasks(self, workflow_id): #noqa not using parameter in mock def restart_task(self, _old_task, new_task): """Create a new task from a failed task checkpoint restart enabled.""" - self.load_task(new_task) + task_state = "WAITING" + self.load_task(new_task, task_state) def finalize_task(self, task): """Set a task's state to completed."""