diff --git a/parastell/nwl_utils.py b/parastell/nwl_utils.py index c539293..44f1162 100644 --- a/parastell/nwl_utils.py +++ b/parastell/nwl_utils.py @@ -145,6 +145,7 @@ def find_coords(data): bounds=[(-np.pi, np.pi)], args=(vmec, wall_s, coords[0], coords[1]), ) + # remap from [-pi,pi] to [0, 2pi] to match parastell's angle convention thetas.append((theta.x[0] + 2 * np.pi) % (2 * np.pi)) return thetas