added pool.close(), pool.join() stop mem leaks

UVAdMIST · Apr 11, 2019 · 523177b · 523177b
1 parent 45dd8bc
commit 523177b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/swmm_mpc/run_ea.py b/swmm_mpc/run_ea.py
@@ -55,6 +55,8 @@ def run_ea(work_dir, config_file, ga_params):
  seed_next_population(hof[0], ga_params.get('nindividuals', 25),
  sm.run.ctl_str_ids, pop_file, sm.run.n_ctl_steps)
  min_cost = min(logbook.select("min"))
+ pool.close()
+ pool.join()
  return hof[0], min_cost