From efa4b9b4e0ea855bb0862eef3349e3b6cd9689f8 Mon Sep 17 00:00:00 2001
From: Teresa Gomez <46339554+teresamg@users.noreply.github.com>
Date: Thu, 8 Dec 2022 15:27:55 -0800
Subject: [PATCH] Removed orig copy of dwdata

---
 src/eddymotion/estimator.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/eddymotion/estimator.py b/src/eddymotion/estimator.py
index 15e7e8bd..a8a86b99 100644
--- 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
                         )