From 5075416b74afb1ba5918599b93cae56fe1de2290 Mon Sep 17 00:00:00 2001 From: leeseungchae <96555334+leeseungchae@users.noreply.github.com> Date: Mon, 10 Jul 2023 17:22:09 +0900 Subject: [PATCH] [modify] multiprocessing --- kss/_utils/multiprocessing.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/kss/_utils/multiprocessing.py b/kss/_utils/multiprocessing.py index 13a147d..586f71d 100644 --- a/kss/_utils/multiprocessing.py +++ b/kss/_utils/multiprocessing.py @@ -30,4 +30,6 @@ def _run_job( else: with mp.Pool(num_workers) as pool: output = pool.map(func, inputs) + pool.close() + pool.join() return output