Removed orig copy of dwdata

nipreps · Dec 15, 2022 · efa4b9b · efa4b9b
1 parent b0d4e85
commit efa4b9b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/eddymotion/estimator.py b/src/eddymotion/estimator.py
@@ -99,8 +99,6 @@ def fit(
         if "num_threads" not in align_kwargs and omp_nthreads is not None:
             align_kwargs["num_threads"] = omp_nthreads
 
-        orig = dwdata
-
         n_iter = len(models)
         for i_iter, model in enumerate(models):
             reg_target_type = (
@@ -138,8 +136,8 @@ def fit(
                         pbar.set_description_str(
                             f"Pass {i_iter + 1}/{n_iter} | Fit and predict b-index <{i}>"
                         )
-                        dwframe = np.asanyarray(orig.dataobj[..., i])
-                        bframe = np.asanyarray(orig.gradients[..., 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
                         )