Apply formatting changes from black v23 (#325)

Author: dnerini

examples/advection_correction.py

@@ -102,7 +102,6 @@ def advection_correction(R, T=5, t=1):
         np.arange(R[0].shape[1], dtype=float), np.arange(R[0].shape[0], dtype=float)
     )
     for i in range(t, T + t, t):
-
         pos1 = (y - i / T * V[1], x - i / T * V[0])
         R1 = map_coordinates(R[0], pos1, order=1)
 

examples/anvil_nowcast.py

@@ -123,6 +123,7 @@
 refobs_field, metadata = utils.to_rainrate(refobs_field[-1], metadata)
 refobs_field[refobs_field < 0.5] = 0.0
 
+
 ###############################################################################
 # Plot the extrapolation, S-PROG and ANVIL nowcasts.
 # --------------------------------------------------

examples/blended_forecast.py

@@ -194,7 +194,6 @@
 leadtimes_min = [30, 60, 90, 120, 150, 180]
 n_leadtimes = len(leadtimes_min)
 for n, leadtime in enumerate(leadtimes_min):
-
     # Nowcast with blending into NWP
     plt.subplot(n_leadtimes, 2, n * 2 + 1)
     plot_precip_field(

examples/data_transformations.py

@@ -70,10 +70,10 @@
 # Test data transformations
 # -------------------------
 
+
 # Define method to visualize the data distribution with boxplots and plot the
 # corresponding skewness
 def plot_distribution(data, labels, skw):
-
     N = len(data)
     fig, ax1 = plt.subplots()
     ax2 = ax1.twinx()

examples/my_first_nowcast.ipynb

@@ -46,7 +46,7 @@
    },
    "outputs": [],
    "source": [
-    "# These libraries are needed for the pygrib library in Colab. \n",
+    "# These libraries are needed for the pygrib library in Colab.\n",
     "# Note that is needed if you install pygrib using pip.\n",
     "# If you use conda, the libraries will be installed automatically.\n",
     "! apt-get install libeccodes-dev libproj-dev\n",
@@ -68,7 +68,7 @@
    "source": [
     "# Uninstall existing shapely\n",
     "# We will re-install shapely in the next step by ignoring the binary\n",
-    "# wheels to make it compatible with other modules that depend on \n",
+    "# wheels to make it compatible with other modules that depend on\n",
     "# GEOS, such as Cartopy (used here).\n",
     "!pip uninstall --yes shapely"
    ]
@@ -83,7 +83,7 @@
    },
    "outputs": [],
    "source": [
-    "# To install cartopy in Colab using pip, we need to install the library \n",
+    "# To install cartopy in Colab using pip, we need to install the library\n",
     "# dependencies first.\n",
     "\n",
     "!apt-get install -qq libgdal-dev libgeos-dev\n",
@@ -170,9 +170,9 @@
    },
    "outputs": [],
    "source": [
-    "# If the configuration file is placed in one of the default locations \n",
-    "# (https://pysteps.readthedocs.io/en/latest/user_guide/set_pystepsrc.html#configuration-file-lookup) \n",
-    "# it will be loaded automatically when pysteps is imported. \n",
+    "# If the configuration file is placed in one of the default locations\n",
+    "# (https://pysteps.readthedocs.io/en/latest/user_guide/set_pystepsrc.html#configuration-file-lookup)\n",
+    "# it will be loaded automatically when pysteps is imported.\n",
     "config_file_path = create_default_pystepsrc(\"pysteps_data\")"
    ]
   },
@@ -198,6 +198,7 @@
    "source": [
     "# Import pysteps and load the new configuration file\n",
     "import pysteps\n",
+    "\n",
     "_ = pysteps.load_config_file(config_file_path, verbose=True)"
    ]
   },
@@ -224,7 +225,8 @@
    "source": [
     "# The default parameters are stored in pysteps.rcparams.\n",
     "from pprint import pprint\n",
-    "pprint(pysteps.rcparams.data_sources['mrms'])"
+    "\n",
+    "pprint(pysteps.rcparams.data_sources[\"mrms\"])"
    ]
   },
   {
@@ -272,7 +274,9 @@
     "start_time = time.time()\n",
     "\n",
     "# Import the data\n",
-    "precipitation, metadata, timestep = load_dataset('mrms',frames=35)  # precipitation in mm/h\n",
+    "precipitation, metadata, timestep = load_dataset(\n",
+    "    \"mrms\", frames=35\n",
+    ")  # precipitation in mm/h\n",
     "\n",
     "end_time = time.time()\n",
     "\n",
@@ -330,7 +334,7 @@
    },
    "outputs": [],
    "source": [
-    "timestep # In minutes"
+    "timestep  # In minutes"
    ]
   },
   {
@@ -450,25 +454,25 @@
     "\n",
     "# Let's define some plotting default parameters for the next plots\n",
     "# Note: This is not strictly needed.\n",
-    "plt.rc('figure', figsize=(4,4))\n",
-    "plt.rc('figure', dpi=100)\n",
-    "plt.rc('font', size=14) # controls default text sizes\n",
-    "plt.rc('axes', titlesize=14) # fontsize of the axes title\n",
-    "plt.rc('axes', labelsize=14) # fontsize of the x and y labels\n",
-    "plt.rc('xtick', labelsize=14) # fontsize of the tick labels\n",
-    "plt.rc('ytick', labelsize=14) # fontsize of the tick labels\n",
+    "plt.rc(\"figure\", figsize=(4, 4))\n",
+    "plt.rc(\"figure\", dpi=100)\n",
+    "plt.rc(\"font\", size=14)  # controls default text sizes\n",
+    "plt.rc(\"axes\", titlesize=14)  # fontsize of the axes title\n",
+    "plt.rc(\"axes\", labelsize=14)  # fontsize of the x and y labels\n",
+    "plt.rc(\"xtick\", labelsize=14)  # fontsize of the tick labels\n",
+    "plt.rc(\"ytick\", labelsize=14)  # fontsize of the tick labels\n",
     "\n",
     "# Let's use the last available composite for nowcasting from the \"training\" data (train_precip[-1])\n",
     "# Also, we will discard any invalid value.\n",
     "valid_precip_values = train_precip[-1][~np.isnan(train_precip[-1])]\n",
     "\n",
     "# Plot the histogram\n",
-    "bins= np.concatenate( ([-0.01,0.01], np.linspace(1,40,39)))\n",
-    "plt.hist(valid_precip_values,bins=bins,log=True, edgecolor='black')\n",
-    "plt.autoscale(tight=True, axis='x')\n",
+    "bins = np.concatenate(([-0.01, 0.01], np.linspace(1, 40, 39)))\n",
+    "plt.hist(valid_precip_values, bins=bins, log=True, edgecolor=\"black\")\n",
+    "plt.autoscale(tight=True, axis=\"x\")\n",
     "plt.xlabel(\"Rainfall intensity [mm/h]\")\n",
     "plt.ylabel(\"Counts\")\n",
-    "plt.title('Precipitation rain rate histogram in mm/h units')\n",
+    "plt.title(\"Precipitation rain rate histogram in mm/h units\")\n",
     "plt.show()"
    ]
   },
@@ -507,11 +511,11 @@
    "source": [
     "from pysteps.utils import transformation\n",
     "\n",
-    "# Log-transform the data to dBR. \n",
+    "# Log-transform the data to dBR.\n",
     "# The threshold of 0.1 mm/h sets the fill value to -15 dBR.\n",
-    "train_precip_dbr, metadata_dbr = transformation.dB_transform(train_precip, metadata, \n",
-    "                                                             threshold=0.1, \n",
-    "                                                             zerovalue=-15.0)"
+    "train_precip_dbr, metadata_dbr = transformation.dB_transform(\n",
+    "    train_precip, metadata, threshold=0.1, zerovalue=-15.0\n",
+    ")"
    ]
   },
   {
@@ -554,7 +558,7 @@
     "\n",
     "# We will only use one composite to fit the function to speed up things.\n",
     "# First, remove the no precip areas.\"\n",
-    "precip_to_fit = valid_precip_dbr[valid_precip_dbr > -15] \n",
+    "precip_to_fit = valid_precip_dbr[valid_precip_dbr > -15]\n",
     "\n",
     "fit_params = scipy.stats.lognorm.fit(precip_to_fit)\n",
     "\n",
@@ -617,7 +621,7 @@
     "plt.title(\"Estimated motion field with the Lukas-Kanade algorithm\")\n",
     "\n",
     "# Plot the last rainfall field in the \"training\" data.\n",
-    "# Remember to use the mm/h precipitation data since plot_precip_field assumes \n",
+    "# Remember to use the mm/h precipitation data since plot_precip_field assumes\n",
     "# mm/h by default. You can change this behavior using the \"units\" keyword.\n",
     "plot_precip_field(train_precip[-1], geodata=metadata, axis=\"off\")\n",
     "\n",
@@ -662,7 +666,7 @@
     "\n",
     "last_observation[~np.isfinite(last_observation)] = metadata[\"zerovalue\"]\n",
     "\n",
-    "# We set the number of leadtimes (the length of the forecast horizon) to the \n",
+    "# We set the number of leadtimes (the length of the forecast horizon) to the\n",
     "# length of the observed/verification preipitation data. In this way, we'll get\n",
     "# a forecast that covers these time intervals.\n",
     "n_leadtimes = observed_precip.shape[0]\n",
@@ -746,7 +750,7 @@
     "    64,\n",
     "]  # In grid points.\n",
     "\n",
-    "scales_in_km = np.array(scales)*4\n",
+    "scales_in_km = np.array(scales) * 4\n",
     "\n",
     "# Set the threshold\n",
     "thr = 1.0  # in mm/h\n",
@@ -764,13 +768,13 @@
     "\n",
     "# Now plot it\n",
     "plt.figure()\n",
-    "x = np.arange(1, n_leadtimes+1) * timestep\n",
+    "x = np.arange(1, n_leadtimes + 1) * timestep\n",
     "plt.plot(x, score, lw=2.0)\n",
     "plt.xlabel(\"Lead time [min]\")\n",
     "plt.ylabel(\"FSS ( > 1.0 mm/h ) \")\n",
     "plt.title(\"Fractions Skill Score\")\n",
     "plt.legend(\n",
-    "    scales_in_km, \n",
+    "    scales_in_km,\n",
     "    title=\"Scale [km]\",\n",
     "    loc=\"center left\",\n",
     "    bbox_to_anchor=(1.01, 0.5),\n",

examples/optical_flow_methods_convergence.py

@@ -116,6 +116,7 @@
 # The "precipitation region" includes the precipitation pattern plus a margin of
 # approximately 20 grid points.
 
+
 ################################################################################
 # Let's create a function to construct different motion fields.
 def create_motion_field(input_precip, motion_type):

examples/plot_linear_blending.py

@@ -210,7 +210,6 @@
 leadtimes_min = [30, 60, 80, 100, 120]
 n_leadtimes = len(leadtimes_min)
 for n, leadtime in enumerate(leadtimes_min):
-
     # Extrapolation
     plt.subplot(n_leadtimes, 4, n * 4 + 1)
     plot_precip_field(

examples/rainfarm_downscale.py

@@ -89,7 +89,6 @@
 # half or double the estimated slope.
 alpha = None
 for n in range(num_realizations):
-
     # Spectral slope estimated from the upscaled field
     precip_hr, alpha = rainfarm.downscale(
         precip_lr, ds_factor=scale_factor, alpha=alpha, return_alpha=True

pysteps/__init__.py

@@ -94,7 +94,6 @@ def config_fname():
 
     file_name = None
     for file_name in _fconfig_candidates_generator():
-
         if file_name is not None:
             if os.path.exists(file_name):
                 st_mode = os.stat(file_name).st_mode

pysteps/blending/linear_blending.py

@@ -41,7 +41,6 @@ def forecast(
     saliency=False,
     nowcast_kwargs=None,
 ):
-
     """Generate a forecast by linearly or saliency-based blending of nowcasts with NWP data
 
     Parameters