From b997ec1fe093bb9ac0481a7c7d4d009d91d9cf45 Mon Sep 17 00:00:00 2001
From: Michael Joseph <josephmje.22@gmail.com>
Date: Fri, 10 Dec 2021 09:03:40 -0500
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: Oscar Esteban <code@oscaresteban.es>
---
 eddymotion/model.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/eddymotion/model.py b/eddymotion/model.py
index fb3a3f34..27a9e310 100644
--- 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}