From 7b01b7a211e1e3c45ff4b325fd460be31c8f49ff Mon Sep 17 00:00:00 2001
From: daniel-dudt <daniel.dudt@princetonstellarators.energy>
Date: Tue, 1 Oct 2024 12:52:45 -0400
Subject: [PATCH] fix bug in QuasisymmetryBoozer.dim_f

---
 desc/objectives/_omnigenity.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/desc/objectives/_omnigenity.py b/desc/objectives/_omnigenity.py
index 415716c2eb..b720608b41 100644
--- 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)