fix bugs

PlasmaControl · Aug 22, 2024 · 0046524 · 0046524
1 parent ed68131
commit 0046524
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/desc/compute/_equil.py b/desc/compute/_equil.py
@@ -872,7 +872,7 @@ def _P_fusion(params, transforms, profiles, data, **kwargs):
             energy = 0.82e6 + 2.45e6  # eV
 
     reaction_rate = jnp.sum(
-        *data["ni"] ** 2
+        data["ni"] ** 2
         * data["<sigma*nu>"]
         * data["sqrt(g)"]
         * transforms["grid"].weights

diff --git a/desc/objectives/__init__.py b/desc/objectives/__init__.py
@@ -11,7 +11,6 @@
     QuadraticFlux,
     ToroidalFlux,
 )
-from ._confinement import HeatingPowerISS04
 from ._equilibrium import (
     CurrentDensity,
     Energy,
@@ -39,6 +38,7 @@
     QuasisymmetryTripleProduct,
     QuasisymmetryTwoTerm,
 )
+from ._power_balance import FusionPower, HeatingPowerISS04
 from ._profiles import Pressure, RotationalTransform, Shear, ToroidalCurrent
 from ._stability import MagneticWell, MercierStability
 from .getters import (