From 910f03bc9f354e721da9e1aac9d8de1050fc5358 Mon Sep 17 00:00:00 2001 From: Casey Jao Date: Wed, 7 Jun 2023 16:53:12 -0400 Subject: [PATCH] Cleanup: fix local and dask receive handling --- covalent/executor/executor_plugins/dask.py | 2 +- covalent/executor/executor_plugins/local.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/covalent/executor/executor_plugins/dask.py b/covalent/executor/executor_plugins/dask.py index e1689e262..d5f22fbd3 100644 --- a/covalent/executor/executor_plugins/dask.py +++ b/covalent/executor/executor_plugins/dask.py @@ -295,7 +295,7 @@ async def receive(self, task_group_metadata: Dict, data: Any) -> List[TaskUpdate terminal_status = RESULT_STATUS.CANCELLED else: received = ReceiveModel.parse_obj(data) - terminal_status = Status(received.status) + terminal_status = Status(received.status.value) for task_id in task_ids: # TODO: Handle the case where the job was cancelled before the task started running diff --git a/covalent/executor/executor_plugins/local.py b/covalent/executor/executor_plugins/local.py index aea207c45..7a246b5b1 100644 --- a/covalent/executor/executor_plugins/local.py +++ b/covalent/executor/executor_plugins/local.py @@ -221,7 +221,7 @@ def _receive(self, task_group_metadata: Dict, data: Any) -> List[TaskUpdate]: terminal_status = RESULT_STATUS.CANCELLED else: received = ReceiveModel.parse_obj(data) - terminal_status = Status(received.status) + terminal_status = Status(received.status.value) task_result = { "dispatch_id": dispatch_id,