reset to include off-diag components

NanoComp · Nov 3, 2021 · 2dde9d6 · 2dde9d6
1 parent d4b1e1a
commit 2dde9d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/adjoint/optimization_problem.py b/python/adjoint/optimization_problem.py
@@ -99,11 +99,11 @@ def __init__(
 
         self.gradient = []
 
-    def __call__(self, rho_vector=None, need_value=True, need_gradient=True):
+    def __call__(self, rho_vector=None, need_value=True, need_gradient=True, beta=None):
         """Evaluate value and/or gradient of objective function.
         """
         if rho_vector:
-            self.update_design(rho_vector=rho_vector)
+            self.update_design(rho_vector=rho_vector, beta=beta)
 
         # Run forward run if requested
         if need_value and self.current_state == "INIT":