frelon_peaksearch: allow to pass kwargs to process_map in `proces…

…s` function (#386) --------- Co-authored-by: noordhee <[email protected]>
FABLE-3DXRD · Jan 22, 2025 · 507440e · 507440e
1 parent 5538a77
commit 507440e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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 = {}