Add a callback function that runs each iteration of the solver

ukaea · Aug 16, 2023 · ce63941 · ce63941
1 parent 3517758
commit ce63941
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pyvmcon/vmcon.py b/src/pyvmcon/vmcon.py
@@ -24,7 +24,7 @@ def solve(
     epsilon: float = 1e-8,
     qsp_options: Optional[Dict[str, Any]] = None,
     initial_B: Optional[np.ndarray] = None,
-    callback: Optional[Callable[[int, np.ndarray, Result], None]],
+    callback: Optional[Callable[[int, np.ndarray, Result], None]] = None,
 ):
     """The main solving loop of the VMCON non-linear constrained optimiser.