black autoformatting

svalinn · Sep 27, 2024 · 8a93251 · 8a93251
1 parent c65130d
commit 8a93251
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/parastell/nwl_utils.py b/parastell/nwl_utils.py
@@ -187,8 +187,10 @@ def flux_coords(plas_eq, wall_s, coords, num_threads):
             for theta_coord in theta_coord_chunk
         ]
     # Ensures theta_coords are all positive (add 360 degrees where needed)
-    theta_coords = [(theta if theta >= 0 else theta + 2 * np.pi) for theta in theta_coords]
-
+    theta_coords = [
+        (theta if theta >= 0 else theta + 2 * np.pi) for theta in theta_coords
+    ]
+
     return phi_coords.tolist(), theta_coords
 
 
@@ -345,7 +347,7 @@ def nwl_plot(
 
     theta_min = 0 - pol_ext / num_theta / 2
     theta_max = pol_ext + pol_ext / num_theta / 2
-    
+
     # Bin particle crossings
     count_mat, phi_bins, theta_bins = np.histogram2d(
         phi_coords,