fix bug in QuasisymmetryBoozer.dim_f

PlasmaControl · Oct 1, 2024 · 7b01b7a · 7b01b7a
1 parent 056fcc4
commit 7b01b7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/desc/objectives/_omnigenity.py b/desc/objectives/_omnigenity.py
@@ -163,7 +163,7 @@ def build(self, use_jit=True, verbose=1):
             M_booz=M_booz,
             N_booz=N_booz,
         )
-        matrix, modes, idx = ptolemy_linear_transform(
+        matrix, _, idx = ptolemy_linear_transform(
             transforms["B"].basis.modes,
             helicity=self.helicity,
             NFP=transforms["B"].basis.NFP,
@@ -180,7 +180,7 @@ def build(self, use_jit=True, verbose=1):
         if verbose > 1:
             timer.disp("Precomputing transforms")
 
-        self._dim_f = idx.size
+        self._dim_f = idx.size * grid.num_rho
 
         if self._normalize:
             scales = compute_scaling_factors(eq)