Suppress sklearn warnings.

dipy · May 10, 2017 · 9df1fe3 · 9df1fe3
1 parent cc4374d
commit 9df1fe3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dipy/reconst/sfm.py b/dipy/reconst/sfm.py
@@ -422,8 +422,10 @@ def fit(self, data, mask=None):
             # parameters at all-zeros, and avoid nasty sklearn errors:
             if not (np.any(~np.isfinite(vox_data)) or np.all(vox_data == 0)):
                 fit_it = vox_data - isopredict[vox]
-                flat_params[vox] = self.solver.fit(self.design_matrix,
-                                                   fit_it).coef_
+                with warnings.catch_warnings():
+                    warnings.simplefilter("ignore")
+                    flat_params[vox] = self.solver.fit(self.design_matrix,
+                                                       fit_it).coef_
 
         if mask is None:
             out_shape = data.shape[:-1] + (-1, )