diff --git a/ImageD11/frelon_peaksearch.py b/ImageD11/frelon_peaksearch.py index a62c78de..e2f28016 100755 --- a/ImageD11/frelon_peaksearch.py +++ b/ImageD11/frelon_peaksearch.py @@ -360,7 +360,7 @@ def guess_bg(ds, scan_number=0, start=0, step=9, n=None): PKCOL = [getattr(ImageD11.cImageD11, p) for p in PKSAVE] -def process(ds, worker_args, ncpu=None): +def process(ds, worker_args, ncpu=None, **process_map_kwargs): """ Runs over the first scan in a dataset in parallel @@ -382,8 +382,9 @@ def process(ds, worker_args, ncpu=None): n_frames = omega.shape[0] args = [(hname, frames_dset, i, omega[i], worker_args) for i in range(n_frames)] + - all_peaks = process_map(pps, args, chunksize=1, max_workers=nthreads) + all_peaks = process_map(pps, args, chunksize=1, max_workers=nthreads, **process_map_kwargs) # make a dict to hold our results in cf_2d_dict = {}