use given iota for normalization scales

PlasmaControl · Aug 14, 2024 · 83ea252 · 83ea252
1 parent db9107b
commit 83ea252
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/desc/objectives/normalization.py b/desc/objectives/normalization.py
@@ -41,7 +41,10 @@ def get_lowest_mode(basis, coeffs):
         scales["A"] = np.pi * scales["a"] ** 2
         scales["V"] = 2 * np.pi * scales["R0"] * scales["A"]
         scales["B_T"] = abs(thing.Psi) / scales["A"]
-        iota_avg = np.mean(np.abs(thing.get_profile("iota")(np.linspace(0, 1, 20))))
+        if thing.iota is not None:
+            iota_avg = np.mean(np.abs(thing.iota(np.linspace(0, 1, 20))))
+        else:
+            iota_avg = np.mean(np.abs(thing.get_profile("iota")(np.linspace(0, 1, 20))))
         if np.isclose(iota_avg, 0):
             scales["B_P"] = scales["B_T"]
         else: