fix the memory leakage bug

bin/bank/pycbc_brute_bank

@@ -259,6 +259,7 @@ class TriangleBank(object):
                 ({k: params[k][idx] for k in params} for idx in range(total_num))
             ):
             waveform_cache += [return_wf]
+        pool.close_pool()
         del pool
 
         for hp in waveform_cache:

pycbc/pool.py

@@ -91,7 +91,7 @@ def allmap(self, fcn, args):
         return results
 
     def map(self, func, items, chunksize=None):
-        """ Catch keyboard interuppts to allow the pool to exit cleanly.
+        """ Catch keyboard interupts to allow the pool to exit cleanly.
 
         Parameters
         ----------
@@ -113,6 +113,13 @@ def map(self, func, items, chunksize=None):
                 self.join()
                 raise KeyboardInterrupt
 
+    def close_pool(self):
+        """ Close the pool and remove the reference
+        """
+        self.close()
+        self.join()
+        atexit.unregister(_shutdown_pool)
+
 def _dummy_broadcast(self, f, args):
     self.map(f, [args] * self.size)
 
@@ -130,6 +137,11 @@ def map(self, f, items):
     imap = map
     imap_unordered = map
 
+    def close_pool(self):
+        ''' Dummy function to be consistent with BroadcastPool
+        '''
+        pass
+
 def use_mpi(require_mpi=False, log=True):
     """ Get whether MPI is enabled and if so the current size and rank
     """