🚸 Terminate early on KeyboardInterrupt (#15)

* 🚸 Terminate early on KeyboardInterrupt

* 👷 Run CI on 3.10

* ⬆️ Bump pytest

* ⬆️ Bump docs dependencies

.github/workflows/CI.yml

@@ -20,7 +20,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: ['3.6', '3.7', '3.8', '3.9', '3.10']
 
     steps:
     - uses: actions/checkout@v2

requirements/ci.txt

@@ -4,6 +4,6 @@ pre-commit~=2.6.0
 pytest-cov~=2.10.1
 pytest-env~=0.6.2
 pytest-sugar~=0.9.4
-pytest~=4.6
+pytest~=6.2
 testfixtures~=6.14.1
 pandas

requirements/docs.txt

@@ -1,3 +1,3 @@
-recommonmark~=0.6
-sphinx-rtd-theme~=0.5
-sphinx~=3.2
+recommonmark~=0.7
+sphinx-rtd-theme~=1.0
+sphinx~=4.2

setup.py

@@ -52,6 +52,7 @@ def read_readme(path):
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
         "Topic :: Software Development :: Libraries :: Python Modules",
         "Topic :: Utilities",
     ],

src/mapply/parallel.py

@@ -79,6 +79,7 @@ def multiprocessing_imap(
 
     Raises:
         Exception: Any error occurred during computation (will terminate the pool early).
+        KeyboardInterrupt: Any KeyboardInterrupt sent by the user (will terminate the pool early).
     """
     n_chunks: Optional[int] = tqdm(iterable, disable=True).__len__()  # doesn't exhaust
     func = partial(func, *args, **kwargs)
@@ -100,10 +101,9 @@ def multiprocessing_imap(
 
     try:
         return list(stage)
-    except Exception:
+    except (Exception, KeyboardInterrupt):
         if pool:
             logger.debug("Terminating ProcessPool")
-            pool.close()
             pool.terminate()
         raise
     finally: