Update examples/benchmarking-scripts/backfill.py

Co-authored-by: Donghee Na <[email protected]>

Author: diegorusso

examples/benchmarking-scripts/backfill.py

@@ -48,16 +48,8 @@ def run_pyperformance(revision):
 
 if __name__ == "__main__":
     original_sigint_handler = signal.signal(signal.SIGINT, signal.SIG_IGN)
-    pool = Pool(8)
     signal.signal(signal.SIGINT, original_sigint_handler)
-    try:
+    with Pool(8) as pool:
         res = pool.map_async(run_pyperformance, get_revisions())
         # Without the timeout this blocking call ignores all signals.
         res.get(86400)
-    except KeyboardInterrupt:
-        print("Caught KeyboardInterrupt, terminating workers")
-        pool.terminate()
-    else:
-        print("Normal termination")
-        pool.close()
-    pool.join()