Apply suggestions from code review

Co-authored-by: Oscar Esteban <[email protected]>
nipreps · Dec 10, 2021 · b997ec1 · b997ec1
1 parent 1ac4ebf
commit b997ec1
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/eddymotion/model.py b/eddymotion/model.py
@@ -55,12 +55,11 @@ def init(gtab, model="DTI", **kwargs):
 
         elif model.lower() in ("sfm", "gp"):
             Model = SparseFascicleModel
-            if model.lower() == "sfm":
-                param = {
-                    "solver": "ElasticNet",
-                    "isotropic": ExponentialIsotropicModel,
-                }
-            else:
+            param = {
+                "solver": "ElasticNet",
+                "isotropic": ExponentialIsotropicModel,
+            }
+            if model.lower() == "gp":
                 from sklearn.gaussian_process import GaussianProcessRegressor
 
                 param = {"solver": GaussianProcessRegressor}