Join Queues

pyiron · Nov 7, 2023 · 5f88f50 · 5f88f50
1 parent 988f546
commit 5f88f50
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pympipool/shared/executorbase.py b/pympipool/shared/executorbase.py
@@ -135,6 +135,7 @@ def execute_parallel_tasks_loop(interface, future_queue):
         if "shutdown" in task_dict.keys() and task_dict["shutdown"]:
             interface.shutdown(wait=task_dict["wait"])
             future_queue.task_done()
+            future_queue.join()
             break
         elif "fn" in task_dict.keys() and "future" in task_dict.keys():
             f = task_dict.pop("future")
@@ -175,6 +176,7 @@ def executor_broker(
                 future_queue.task_done()
             else:
                 future_queue.task_done()
+                future_queue.join()
                 break