Fixed logo_split() call and dwdata->data

nipreps · Mar 27, 2024 · a3a3f1b · a3a3f1b
1 parent 88bf1dc
commit a3a3f1b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/src/eddymotion/data/splitting.py b/src/eddymotion/data/splitting.py
@@ -47,7 +47,7 @@ def lovo_split(data, index):
     """
 
     # if the size of the mask does not match data, cache is stale
-    mask = np.zeros(len(dwdata), dtype=bool)
+    mask = np.zeros(len(data), dtype=bool)
     mask[index] = True
 
     train_data = data.dataobj[..., ~mask]

diff --git a/src/eddymotion/estimator.py b/src/eddymotion/estimator.py
@@ -151,11 +151,7 @@ def fit(
                         pbar.set_description_str(
                             f"Pass {i_iter + 1}/{n_iter} | Fit and predict b-index <{i}>"
                         )
-                        dwframe = np.asanyarray(dwdata.dataobj[..., i])
-                        bframe = np.asanyarray(dwdata.gradients[..., i])
-                        data_train, data_test = logo_split(
-                            dwdata, dwframe, bframe, i, with_b0=True
-                        )
+                        data_train, data_test = logo_split(dwdata, i)
 
                         grad_str = f"{i}, {data_test[1][:3]}, b={int(data_test[1][3])}"
                         pbar.set_description_str(f"[{grad_str}], {n_jobs} jobs")