fixing more tests 2

PlasmaControl · Sep 28, 2024 · 2ace1f6 · 2ace1f6
1 parent 392c9e9
commit 2ace1f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/desc/compute/_profiles.py b/desc/compute/_profiles.py
@@ -9,7 +9,7 @@
 expensive computations.
 """
 
-from quadax import simpson
+from quadax import cumulative_simpson
 from scipy.constants import elementary_charge, mu_0
 
 from desc.backend import cond, jnp
@@ -143,7 +143,7 @@ def _chi_r(params, transforms, profiles, data, **kwargs):
     resolution_requirement="r",
 )
 def _chi(params, transforms, profiles, data, **kwargs):
-    chi = simpson(
+    chi = cumulative_simpson(
         y=transforms["grid"].compress(data["chi_r"]),
         x=transforms["grid"].compress(data["rho"]),
         initial=0,