Linted

hiddenSymmetries · Oct 3, 2023 · b03fa7d · b03fa7d
1 parent 9d1f036
commit b03fa7d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/examples/2_Intermediate/stage_two_optimization_planar_coils.py b/examples/2_Intermediate/stage_two_optimization_planar_coils.py
@@ -110,7 +110,7 @@
 # Define the individual terms objective function:
 Jf = SquaredFlux(s, bs)
 Jls = [CurveLength(c) for c in base_curves]
-Jccdist = CurveCurveDistance(curves, CC_THRESHOLD, num_basecurves=ncoils) #Error
+Jccdist = CurveCurveDistance(curves, CC_THRESHOLD, num_basecurves=ncoils)  # Error
 Jcsdist = CurveSurfaceDistance(curves, s, CS_THRESHOLD)
 Jcs = [LpCurveCurvature(c, 2, CURVATURE_THRESHOLD) for c in base_curves]
 Jmscs = [MeanSquaredCurvature(c) for c in base_curves]
@@ -131,6 +131,7 @@
 # are not in least-squares form, so we write a little wrapper function that we
 # pass directly to scipy.optimize.minimize
 
+
 def fun(dofs):
     JF.x = dofs
     J = JF.J()

diff --git a/src/simsopt/geo/curve.py b/src/simsopt/geo/curve.py
@@ -883,6 +883,7 @@ def create_equally_spaced_curves(ncurves, nfp, stellsym, R0=1.0, R1=0.5, order=6
         curves.append(curve)
     return curves
 
+
 def create_equally_spaced_planar_curves(ncurves, nfp, stellsym, R0=1.0, R1=0.5, order=6, numquadpoints=None):
     """
     Create ``ncurves`` curves of type
@@ -906,7 +907,7 @@ def create_equally_spaced_planar_curves(ncurves, nfp, stellsym, R0=1.0, R1=0.5,
         center = [R0 * cos(angle), R0 * sin(angle), 0]
         rotation = [1, -cos(angle), -sin(angle), 0]
         dofs = np.zeros(len(curve.get_dofs()))
-    
+
         j = 0
         for i in rcCoeffs:
             dofs[j] = i

diff --git a/src/simsopt/geo/curveplanarfourier.py b/src/simsopt/geo/curveplanarfourier.py
@@ -23,7 +23,7 @@ class CurvePlanarFourier(sopp.CurvePlanarFourier, Curve):
 
     .. math::
        [r_{c,0}, \cdots, r_{c,\text{order}}, r_{s,1}, \cdots, r_{s,\text{order}}, q_0, q_i, q_j, q_k, x_{\text{center}}, y_{\text{center}}, z_{\text{center}}]
-    
+
 
     """
 

diff --git a/tests/geo/test_curve.py b/tests/geo/test_curve.py
@@ -100,6 +100,7 @@ def get_curve(curvetype, rotated, x=np.asarray([0.5])):
 class Testing(unittest.TestCase):
 
     curvetypes = ["CurveXYZFourier", "JaxCurveXYZFourier", "CurveRZFourier", "CurvePlanarFourier", "CurveHelical", "CurveHelicalInitx0"]
+
     def test_curve_helical_xyzfourier(self):
         x = np.asarray([0.6])
         curve1 = CurveHelical(x, 2, 5, 2, 1.0, 0.3)