Fixing Nfp periodicity in BoozerAnalytic.

hiddenSymmetries · Jan 23, 2024 · 3585880 · 3585880
1 parent db10d50
commit 3585880
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/simsopt/field/boozermagneticfield.py b/src/simsopt/field/boozermagneticfield.py
@@ -246,7 +246,7 @@ def _modB_impl(self, modB):
         zetas = points[:, 2]
         psi = s*self.psi0
         r = np.sqrt(np.abs(2*psi/self.Bbar))
-        modB[:, 0] = self.B0*(1 + self.etabar*r*np.cos(thetas-self.N*zetas)) + self.B0z*np.cos(zetas)
+        modB[:, 0] = self.B0*(1 + self.etabar*r*np.cos(thetas-self.N*zetas)) + self.B0z*np.cos(self.N*zetas)
 
     def _dmodBds_impl(self, dmodBds):
         points = self.get_points_ref()
@@ -279,7 +279,7 @@ def _dmodBdzeta_impl(self, dmodBdzeta):
         zetas = points[:, 2]
         psi = s*self.psi0
         r = np.sqrt(np.abs(2*psi/self.Bbar))
-        dmodBdzeta[:, 0] = self.N*self.B0*self.etabar*r*np.sin(thetas-self.N*zetas) - self.B0z*np.sin(zetas)
+        dmodBdzeta[:, 0] = self.N*self.B0*self.etabar*r*np.sin(thetas-self.N*zetas) - self.N*self.B0z*np.sin(self.N*zetas)
 
     def _K_impl(self, K):
         points = self.get_points_ref()