Update Wrappers/Python/cil/optimisation/functions/TotalVariation.py

Co-authored-by: Gemma Fardell <47746591+gfardell@users.noreply.github.com> Signed-off-by: Margaret Duff <43645617+MargaretDuff@users.noreply.github.com>
TomographicImaging · Aug 9, 2023 · 8698a5a · 8698a5a
1 parent 2d6a48d
commit 8698a5a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Wrappers/Python/cil/optimisation/functions/TotalVariation.py b/Wrappers/Python/cil/optimisation/functions/TotalVariation.py
@@ -368,8 +368,8 @@ def _fista_on_dual_rof(self, x, tau, out = None):
             tmp = p1
             p1 = p2
             p2 = tmp
-            if self.warmstart:
-                self.p2=p2
+        if self.warmstart:
+            self._p2 = p2
 
         # Print stopping information (iterations and tolerance error) of FGP_TV     
         if self.info: