Fix kwargs dict retrieval

hiddenSymmetries · Dec 18, 2024 · 8b403a2 · 8b403a2
1 parent d6bea2b
commit 8b403a2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/simsopt/solve/permanent_magnet_optimization.py b/src/simsopt/solve/permanent_magnet_optimization.py
@@ -200,9 +200,10 @@ def relax_and_split(pm_opt, m0=None, **kwargs):
 
     # set the nonconvex step in the algorithm
     reg_rs = 0.0
-    nu = kwargs.pop("nu", 1e100)
-    reg_l0 = kwargs.pop("reg_l0", 0.0)
-    reg_l1 = kwargs.pop("reg_l1", 0.0)
+    nu = kwargs.get("nu", 1e100)
+    reg_l0 = kwargs.get("reg_l0", 0.0)
+    reg_l1 = kwargs.get("reg_l1", 0.0)
+
     max_iter_RS = kwargs.pop('max_iter_RS', 1)
     epsilon_RS = kwargs.pop('epsilon_RS', 1e-3)