Minor fix in TAOSolver.solve

tlm-adjoint · Jul 19, 2024 · 33cda51 · 33cda51
1 parent a693934
commit 33cda51
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tlm_adjoint/optimization.py b/tlm_adjoint/optimization.py
@@ -522,10 +522,10 @@ def solve(self, M):
         x = self._vec_interface.new_vec()
         x.to_petsc(M)
 
-        self._M[0] = tuple(var_new(m, static=var_is_static(m),
-                                   cache=var_is_cached(m))
-                           for m in M)
         try:
+            self._M[0] = tuple(var_new(m, static=var_is_static(m),
+                                       cache=var_is_cached(m))
+                               for m in M)
             self.tao.solve(x.vec)
         finally:
             self._M[0] = None