From c4d3bd4c815abf6b88d0371b0d8b6b73be8d1d01 Mon Sep 17 00:00:00 2001
From: mj-will <michaeljw1@googlemail.com>
Date: Wed, 20 Nov 2024 11:47:31 +0000
Subject: [PATCH 1/2] pass kwargs to pool constructor in choose_pool

---
 pycbc/pool.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/pycbc/pool.py b/pycbc/pool.py
index bffa2202689..5cb6b0a3419 100644
--- a/pycbc/pool.py
+++ b/pycbc/pool.py
@@ -131,6 +131,10 @@ def _dummy_broadcast(self, f, args):
     self.map(f, [args] * self.size)
 
 class SinglePool(object):
+
+    def __init__(self, **_):
+        pass
+
     def broadcast(self, fcn, args):
         return self.map(fcn, [args])
 
@@ -174,15 +178,18 @@ def use_mpi(require_mpi=False, log=True):
     return use_mpi, size, rank
 
 
-def choose_pool(processes, mpi=False):
-    """ Get processing pool
+def choose_pool(processes, mpi=False, **kwargs):
+    """ Get processing pool.
+
+    Keyword arguments are pass to the pool constructor.
     """
     do_mpi, size, rank = use_mpi(require_mpi=mpi)
     if do_mpi:
         try:
             import schwimmbad
             pool = schwimmbad.choose_pool(mpi=do_mpi,
-                                          processes=(size - 1))
+                                          processes=(size - 1),
+                                          **kwargs)
             pool.broadcast = types.MethodType(_dummy_broadcast, pool)
             atexit.register(pool.close)
 
@@ -197,11 +204,11 @@ def choose_pool(processes, mpi=False):
             raise ValueError("Failed to start up an MPI pool, "
                              "install mpi4py / schwimmbad")
     elif processes == 1:
-        pool = SinglePool()
+        pool = SinglePool(**kwargs)
     else:
         if processes == -1:
             processes = cpu_count()
-        pool = BroadcastPool(processes)
+        pool = BroadcastPool(processes, **kwargs)
 
     pool.size = processes
     if size:

From 04893faeea0f10ba4deac45473cd704a0d035b43 Mon Sep 17 00:00:00 2001
From: mj-will <michaeljw1@googlemail.com>
Date: Wed, 20 Nov 2024 12:06:02 +0000
Subject: [PATCH 2/2] fix typo in doc-string

---
 pycbc/pool.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pycbc/pool.py b/pycbc/pool.py
index 5cb6b0a3419..1e3539eea1d 100644
--- a/pycbc/pool.py
+++ b/pycbc/pool.py
@@ -181,7 +181,7 @@ def use_mpi(require_mpi=False, log=True):
 def choose_pool(processes, mpi=False, **kwargs):
     """ Get processing pool.
 
-    Keyword arguments are pass to the pool constructor.
+    Keyword arguments are passed to the pool constructor.
     """
     do_mpi, size, rank = use_mpi(require_mpi=mpi)
     if do_mpi: