Skip to content

Commit

Permalink
RF: Set dask as the default paralellization engine.
Browse files Browse the repository at this point in the history
  • Loading branch information
arokem committed Aug 25, 2023
1 parent 3651cef commit 4e8ca6e
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions AFQ/segmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
-------
Expand Down

0 comments on commit 4e8ca6e

Please sign in to comment.