add ExponentialIsotropicModel if multi-shell

nipreps · Dec 10, 2021 · 47677f3 · 47677f3
1 parent 51034a5
commit 47677f3
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/eddymotion/model.py b/eddymotion/model.py
@@ -55,10 +55,14 @@ def init(gtab, model="DTI", **kwargs):
 
         elif model.lower() in ("sfm", "gp"):
             Model = SparseFascicleModel
-            param = {
-                "solver": "ElasticNet",
-                "isotropic": ExponentialIsotropicModel,
-            }
+            param = {"solver": "ElasticNet"}
+
+            from dipy.core.gradients import check_multi_b
+            multi_b = check_multi_b(gtab, 2, non_zero=False)
+            if multi_b:
+                from dipy.reconst.sfm import ExponentialIsotropicModel
+                param["isotropic"].append(ExponentialIsotropicModel)
+
             if model.lower() == "gp":
                 from sklearn.gaussian_process import GaussianProcessRegressor