Format with latest version of black

pySTEPS · Mar 5, 2024 · 3167a11 · 3167a11
1 parent be8eea4
commit 3167a11
Show file tree

Hide file tree

Showing 11 changed files with 22 additions and 25 deletions.
diff --git a/doc/source/conf.py b/doc/source/conf.py
@@ -217,7 +217,7 @@ def set_root():
 latex_elements = {
     "papersize": "a4paper",
     "pointsize": "10pt",
-    "preamble": latex_preamble
+    "preamble": latex_preamble,
     # Latex figure (float) alignment
     #
     # 'figure_align': 'htbp',

diff --git a/pysteps/decorators.py b/pysteps/decorators.py
@@ -228,9 +228,9 @@ def _interpolator_with_preamble(xy_coord, values, xgrid, ygrid, **kwargs):
                 indy = 0
                 for subygrid in subygrids:
                     deltay = subygrid.size
-                    interpolated[
-                        :, indy : (indy + deltay), indx : (indx + deltax)
-                    ] = interpolator(xy_coord, values, subxgrid, subygrid, **kwargs)
+                    interpolated[:, indy : (indy + deltay), indx : (indx + deltax)] = (
+                        interpolator(xy_coord, values, subxgrid, subygrid, **kwargs)
+                    )
                     indy += deltay
                 indx += deltax
 

diff --git a/pysteps/io/importers.py b/pysteps/io/importers.py
@@ -1366,9 +1366,9 @@ def import_odim_hdf5(filename, qty="RATE", **kwargs):
                                     mask = np.logical_and(~mask_u, ~mask_n)
                                     quality = np.empty(arr.shape)  # , dtype=float)
                                     quality[mask] = arr[mask] * gain + offset
-                                    quality[
-                                        ~mask
-                                    ] = np.nan  # a qui -----------------------------
+                                    quality[~mask] = (
+                                        np.nan
+                                    )  # a qui -----------------------------
 
     if precip is None:
         raise IOError("requested quantity %s not found" % qty)

diff --git a/pysteps/noise/fftgenerators.py b/pysteps/noise/fftgenerators.py
@@ -708,9 +708,7 @@ def initialize_nonparam_2d_nested_filter(field, gridres=1.0, **kwargs):
         # update indices
         level += 1
         Idxi, Idxj = _split_field((0, dim[0]), (0, dim[1]), 2**level)
-        Idxipsd, Idxjpsd = _split_field(
-            (0, 2**max_level), (0, 2**max_level), 2**level
-        )
+        Idxipsd, Idxjpsd = _split_field((0, 2**max_level), (0, 2**max_level), 2**level)
 
     return {"field": F, "input_shape": field.shape[1:], "use_full_fft": True}
 

diff --git a/pysteps/nowcasts/linda.py b/pysteps/nowcasts/linda.py
@@ -572,8 +572,7 @@ def _compute_window_weights(coords, grid_height, grid_width, window_radius):
             dx = c[1] - grid_x
 
             w[i, :] = np.exp(
-                -dy * dy / (2 * window_radius_1**2)
-                - dx * dx / (2 * window_radius_2**2)
+                -dy * dy / (2 * window_radius_1**2) - dx * dx / (2 * window_radius_2**2)
             )
     else:
         w[0, :] = np.ones((grid_height, grid_width))

diff --git a/pysteps/nowcasts/sseps.py b/pysteps/nowcasts/sseps.py
@@ -729,12 +729,12 @@ def worker(j):
                                 else:
                                     EPS_ = None
                                 # apply AR(p) process to cascade level
-                                precip_cascades[
-                                    i, :, :, :
-                                ] = autoregression.iterate_ar_model(
-                                    precip_cascades[i, :, :, :],
-                                    phi[m, n, i, :],
-                                    eps=EPS_,
+                                precip_cascades[i, :, :, :] = (
+                                    autoregression.iterate_ar_model(
+                                        precip_cascades[i, :, :, :],
+                                        phi[m, n, i, :],
+                                        eps=EPS_,
+                                    )
                                 )
                                 EPS_ = None
                             rc[m][n][j] = precip_cascades.copy()

diff --git a/pysteps/scripts/fit_vel_pert_params.py b/pysteps/scripts/fit_vel_pert_params.py
@@ -51,9 +51,7 @@
     mu = dp_perp_sum / dp_perp_n
 
     std_perp.append(
-        np.sqrt(
-            (dp_perp_sq_sum - 2 * mu * dp_perp_sum + dp_perp_n * mu**2) / dp_perp_n
-        )
+        np.sqrt((dp_perp_sq_sum - 2 * mu * dp_perp_sum + dp_perp_n * mu**2) / dp_perp_n)
     )
 
 try:

diff --git a/pysteps/tests/helpers.py b/pysteps/tests/helpers.py
@@ -4,6 +4,7 @@
 
 Collection of helper functions for the testing suite.
 """
+
 from datetime import datetime
 
 import numpy as np

diff --git a/pysteps/tests/test_motion.py b/pysteps/tests/test_motion.py
@@ -349,7 +349,7 @@ def test_vet_padding():
             verbose=False,
             sectors=((16, 4, 2), (16, 4, 2)),
             options=dict(maxiter=5),
-            padding=padding
+            padding=padding,
             # We use only a few iterations since
             # we don't care about convergence in this test
         )

diff --git a/pysteps/utils/interface.py b/pysteps/utils/interface.py
@@ -205,9 +205,9 @@ def donothing(R, metadata=None, *args, **kwargs):
     methods_objects["rm_rdisc"] = spectral.remove_rain_norain_discontinuity
 
     # tapering methods
-    methods_objects[
-        "compute_mask_window_function"
-    ] = tapering.compute_mask_window_function
+    methods_objects["compute_mask_window_function"] = (
+        tapering.compute_mask_window_function
+    )
     methods_objects["compute_window_function"] = tapering.compute_window_function
 
     # transformation methods

diff --git a/pysteps/visualization/utils.py b/pysteps/visualization/utils.py
@@ -13,6 +13,7 @@
     get_geogrid
     get_basemap_axis
 """
+
 import warnings
 
 import matplotlib.pylab as plt