diff --git a/brainiak/fcma/voxelselector.py b/brainiak/fcma/voxelselector.py index 16f086918..5fa4304da 100644 --- a/brainiak/fcma/voxelselector.py +++ b/brainiak/fcma/voxelselector.py @@ -28,7 +28,7 @@ from . import fcma_extension from . import cython_blas as blas import logging -import pathos.multiprocessing +import multiprocess logger = logging.getLogger(__name__) @@ -412,11 +412,9 @@ def _cross_validation_for_one_voxel(vid, num_folds, inlist = [(i + task[0], self.num_folds, data[i, :, :], self.labels) for i in range(task[1])] - pool = pathos.multiprocessing.ProcessingPool(None) - results = list(pool.map( - lambda x: _cross_validation_for_one_voxel - (x[0], x[1], x[2], x[3]), - inlist)) + with multiprocess.Pool(None) as pool: + results = list(pool.starmap(_cross_validation_for_one_voxel, + inlist)) else: results = [] for i in range(task[1]):