diff --git a/desc/objectives/_stability.py b/desc/objectives/_stability.py index b65c931ab2..bf448a29b1 100644 --- a/desc/objectives/_stability.py +++ b/desc/objectives/_stability.py @@ -94,11 +94,12 @@ def build(self, eq=None, use_jit=True, verbose=1): eq = eq or self._eq if self._grid is None: grid = LinearGrid( + L=eq.L_grid, M=eq.M_grid, N=eq.N_grid, NFP=eq.NFP, sym=eq.sym, - rho=np.linspace(1 / 5, 1, 5), + axis=False, ) else: grid = self._grid @@ -296,11 +297,12 @@ def build(self, eq=None, use_jit=True, verbose=1): eq = eq or self._eq if self._grid is None: grid = LinearGrid( + L=eq.L_grid, M=eq.M_grid, N=eq.N_grid, NFP=eq.NFP, sym=eq.sym, - rho=np.linspace(1 / 5, 1, 5), + axis=False, ) else: grid = self._grid diff --git a/tests/test_optimizer.py b/tests/test_optimizer.py index ad7b2af22b..cccfe962e4 100644 --- a/tests/test_optimizer.py +++ b/tests/test_optimizer.py @@ -874,7 +874,6 @@ def test_constrained_AL_lsq(): constraints = ( FixBoundaryR(eq=eq, modes=[0, 0, 0]), # fix specified major axis position - FixBoundaryZ(eq=eq), # fix Z shape but not R FixPressure(eq=eq), # fix pressure profile FixIota(eq=eq), # fix rotational transform profile FixPsi(eq=eq), # fix total toroidal magnetic flux @@ -917,7 +916,6 @@ def test_constrained_AL_lsq(): @pytest.mark.slow @pytest.mark.regression -@pytest.mark.xfail def test_constrained_AL_scalar(): """Tests that the augmented Lagrangian constrained optimizer does something.""" eq = desc.examples.get("SOLOVEV")