]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
GH-83658: make multiprocessing.Pool raise an exception if maxtasksperchild is not...
authorIrit Katriel <1055913+iritkatriel@users.noreply.github.com>
Fri, 17 Jun 2022 07:14:26 +0000 (08:14 +0100)
committerGitHub <noreply@github.com>
Fri, 17 Jun 2022 07:14:26 +0000 (00:14 -0700)
Closes #83658.

Lib/multiprocessing/pool.py
Lib/test/_test_multiprocessing.py
Misc/NEWS.d/next/Library/2022-05-30-21-42-50.gh-issue-83658.01Ntx0.rst [new file with mode: 0644]

index bbe05a550c349cdaaf32f2d849a5b56aaed9a583..961d7e5991847af1ea8034a5a37d9159cf4614ae 100644 (file)
@@ -203,6 +203,9 @@ class Pool(object):
             processes = os.cpu_count() or 1
         if processes < 1:
             raise ValueError("Number of processes must be at least 1")
+        if maxtasksperchild is not None:
+            if not isinstance(maxtasksperchild, int) or maxtasksperchild <= 0:
+                raise ValueError("maxtasksperchild must be a positive int or None")
 
         if initializer is not None and not callable(initializer):
             raise TypeError('initializer must be a callable')
index b20bc0b08015d34d17454828a501afc4e5bd3f16..75969975af35ba0f3b518fd28eeaabb6c10c7976 100644 (file)
@@ -2863,6 +2863,11 @@ class _TestPoolWorkerLifetime(BaseTestCase):
         for (j, res) in enumerate(results):
             self.assertEqual(res.get(), sqr(j))
 
+    def test_pool_maxtasksperchild_invalid(self):
+        for value in [0, -1, 0.5, "12"]:
+            with self.assertRaises(ValueError):
+                multiprocessing.Pool(3, maxtasksperchild=value)
+
     def test_worker_finalization_via_atexit_handler_of_multiprocessing(self):
         # tests cases against bpo-38744 and bpo-39360
         cmd = '''if 1:
diff --git a/Misc/NEWS.d/next/Library/2022-05-30-21-42-50.gh-issue-83658.01Ntx0.rst b/Misc/NEWS.d/next/Library/2022-05-30-21-42-50.gh-issue-83658.01Ntx0.rst
new file mode 100644 (file)
index 0000000..a187309
--- /dev/null
@@ -0,0 +1 @@
+Make :class:`multiprocessing.Pool` raise an exception if ``maxtasksperchild`` is not ``None`` or a positive int.