Merge pull request #344 from jons-pf/patch-1

[FIX] theta_vmec computation in vmec_fieldlines

src/simsopt/mhd/vmec_diagnostics.py

@@ -1306,7 +1306,7 @@ def residual(theta_v, phi0, theta_p_target, jradius):
         This function is used for computing an array of values of theta_vmec that
         give a desired theta_pest array.
         """
-        return theta_p_target - (theta_v + np.sum(lmns[js, :, None] * np.sin(xm[:, None] * theta_v - xn[:, None] * phi0), axis=0))
+        return theta_p_target - (theta_v + np.sum(lmns[jradius, :, None] * np.sin(xm[:, None] * theta_v - xn[:, None] * phi0), axis=0))
 
     theta_vmec = np.zeros((ns, nalpha, nl))
     for js in range(ns):