release actor lock when set_subtask_result (#3210)

mars-project · Aug 9, 2022 · 14c9339 · 14c9339
1 parent bfb6e73
commit 14c9339
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mars/services/task/supervisor/task.py b/mars/services/task/supervisor/task.py
@@ -415,7 +415,7 @@ async def set_subtask_result(self, subtask_result: SubtaskResult):
             "Set subtask %s with result %s.", subtask_result.subtask_id, subtask_result
         )
         if self._cur_processor is not None:
-            await self._cur_processor.set_subtask_result(subtask_result)
+            yield self._cur_processor.set_subtask_result(subtask_result)
 
     def is_done(self) -> bool:
         for processor in self._task_id_to_processor.values():