Update src/eddymotion/estimator.py

Co-authored-by: Oscar Esteban <[email protected]>
nipreps · Mar 28, 2024 · 9591b81 · 9591b81
1 parent 14cd4d0
commit 9591b81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/eddymotion/estimator.py b/src/eddymotion/estimator.py
@@ -151,7 +151,7 @@ def fit(
                         pbar.set_description_str(
                             f"Pass {i_iter + 1}/{n_iter} | Fit and predict b-index <{i}>"
                         )
-                        data_train, data_test = lovo_split(dwdata, i)
+                        data_train, data_test = lovo_split(dwdata, i, with_b0=True)
                         grad_str = f"{i}, {data_test[1][:3]}, b={int(data_test[1][3])}"
                         pbar.set_description_str(f"[{grad_str}], {n_jobs} jobs")