Merge pull request #1009 from arokem/dask_default

RF: Set dask as the default paralellization engine.
yeatmanlab · Aug 28, 2023 · a774fb8 · a774fb8
2 parents 3651cef + 4e8ca6e
commit a774fb8
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/AFQ/segmentation.py b/AFQ/segmentation.py
@@ -119,8 +119,7 @@ def __init__(self,
                  seg_algo='AFQ',
                  clip_edges=False,
                  parallel_segmentation={
-                     "n_jobs": 4, "engine": "joblib",
-                     "backend": "loky"},
+                     "n_jobs": 4, "engine": "dask"},
                  progressive=True,
                  greater_than=50,
                  rm_small_clusters=50,
@@ -340,7 +339,7 @@ def segment(self, bundle_dict, tg, mapping, fdata=None, fbval=None,
                     'end': img5}}
         tg : StatefulTractogram
             Bundles to segment
-        mapping : DiffeomorphicMap, or equivalent interface 
+        mapping : DiffeomorphicMap, or equivalent interface
             A mapping between DWI space and a template.
         fdata, fbval, fbvec : str
             Full path to data, bvals, bvecs
@@ -1099,7 +1098,7 @@ def sl_curve_dist(curve1, curve2):
     Parameters
     ----------
     curve1, curve2 : 2d array-like
-        Two curves calculated from sl_curve. 
+        Two curves calculated from sl_curve.
 
     Returns
     -------