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,