fix bug in serial server

funkelab · Oct 25, 2024 · ea7089f · ea7089f
1 parent 41d7de4
commit ea7089f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/daisy/serial_server.py b/daisy/serial_server.py
@@ -62,6 +62,7 @@ def run_blockwise(self, tasks: list[Task], scheduler=None) -> dict[str, TaskStat
                     del process_funcs[block.task_id]
 
             if len(process_funcs) == 0:
-                self.notify_server_exit()
-                return scheduler.task_states
-        raise NotImplementedError("Unreachable")
+                break
+
+        self.notify_server_exit()
+        return scheduler.task_states