From 8a93251124604aba04e81f6275df502ac5b8a1da Mon Sep 17 00:00:00 2001 From: Edgar Date: Thu, 26 Sep 2024 20:41:57 -0500 Subject: [PATCH] black autoformatting --- parastell/nwl_utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/parastell/nwl_utils.py b/parastell/nwl_utils.py index 4c4e774..0996fa2 100644 --- 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,