Better hygiene in use of Pool in run_regression (castorini#1743)

dannychn11 · Jan 19, 2022 · d7a5d6c · d7a5d6c
1 parent 66ac36b
commit d7a5d6c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/python/run_regression.py b/src/main/python/run_regression.py
@@ -228,7 +228,7 @@ def run_search(cmd):
             for cmd in search_cmds:
                 logger.info(' '.join(cmd))
         else:
-            p = Pool(args.search_pool)
-            p.map(run_search, search_cmds)
+            with Pool(args.search_pool) as p:
+                p.map(run_search, search_cmds)
 
         evaluate_and_verify(yaml_data, args.dry_run)