diff --git a/desc/objectives/_neoclassical.py b/desc/objectives/_neoclassical.py index 0306401c38..2936fe6ebb 100644 --- a/desc/objectives/_neoclassical.py +++ b/desc/objectives/_neoclassical.py @@ -173,12 +173,11 @@ def build(self, use_jit=True, verbose=1): ) else: rho = self._grid_1dr.compress(self._grid_1dr.nodes[:, 0]) - self.constants["rho"] = rho - self.constants["quad"] = get_quadrature( + self._constants["rho"] = rho + self._constants["quad"] = get_quadrature( leggauss_lob(self._hyperparameters.pop("num_quad")), Bounce1D._default_automorphism, ) - self._dim_f = rho.size self._target, self._bounds = _parse_callable_target_bounds( self._target, self._bounds, rho