Merge pull request #55 from BlockScience/staging

restart threads
2019-08-27 15:28:54 -04:00 · 2019-08-27 15:28:54 -04:00 · 4f58a169c5
parent de9a708d43 db7de4fe4f
commit 4f58a169c5
1 changed files with 4 additions and 1 deletions
--- a/cadCAD/engine/simulation.py
+++ b/cadCAD/engine/simulation.py
@ -222,10 +222,13 @@ class Executor:

            return first_timestep_per_run

+        tp = TPool(runs)
        pipe_run: List[List[Dict[str, Any]]] = flatten(
-            TPool().map(
+            tp.map(
                lambda run: execute_run(sweep_dict, states_list, configs, env_processes, time_seq, run),
                list(range(runs))
            )
        )
+
+        tp.clear()
        return pipe_run