Cleanup: fix local and dask receive handling

AgnostiqHQ · Jun 7, 2023 · 910f03b · 910f03b
1 parent 884a530
commit 910f03b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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,