add missing lambda function to iota_psi

PlasmaControl · Aug 23, 2023 · 092d6a3 · 092d6a3
1 parent 2a8f941
commit 092d6a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/desc/compute/_profiles.py b/desc/compute/_profiles.py
@@ -1282,7 +1282,7 @@ def _iota_psi(params, transforms, profiles, data, **kwargs):
     # Assume iota may be expanded as an even power series of ρ so that this
     # condition is satisfied.
     data["iota_psi"] = transforms["grid"].replace_at_axis(
-        data["iota_r"] / data["psi_r"], data["iota_rr"] / data["psi_rr"]
+        data["iota_r"] / data["psi_r"], lambda: data["iota_rr"] / data["psi_rr"]
     )
     return data