From 8b403a2d3371db5b93ab22ba29408b630ea9f116 Mon Sep 17 00:00:00 2001 From: ryanwu4 Date: Wed, 18 Dec 2024 18:19:11 -0500 Subject: [PATCH] Fix kwargs dict retrieval --- src/simsopt/solve/permanent_magnet_optimization.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/simsopt/solve/permanent_magnet_optimization.py b/src/simsopt/solve/permanent_magnet_optimization.py index 851476047..a654def87 100644 --- 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)