earthobservations · gutzbenj · Mar 11, 2024 · Mar 6, 2024 · Mar 6, 2024 · Mar 6, 2024
diff --git a/benchmarks/interpolation.py b/benchmarks/interpolation.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 import datetime as dt
@@ -48,7 +47,12 @@ class Data:
 
 
 def request_weather_data(
-    parameter: str, lat: float, lon: float, distance: float, start_date: dt.datetime, end_date: dt.datetime
+    parameter: str,
+    lat: float,
+    lon: float,
+    distance: float,
+    start_date: dt.datetime,
+    end_date: dt.datetime,
 ):
     stations = DwdObservationRequest(
         parameter=parameter,

diff --git a/benchmarks/interpolation_over_time.py b/benchmarks/interpolation_over_time.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 import os
@@ -45,13 +44,15 @@ def get_regular_df(parameter: str, start_date: datetime, end_date: datetime, exc
 
 def get_rmse(regular_values: pl.Series, interpolated_values: pl.Series) -> float:
     return root_mean_squared_error(
-        regular_values.reshape((-1, 1)).to_list(), interpolated_values.reshape((-1, 1)).to_list()
+        regular_values.reshape((-1, 1)).to_list(),
+        interpolated_values.reshape((-1, 1)).to_list(),
     )
 
 
 def get_corr(regular_values: pl.Series, interpolated_values: pl.Series) -> float:
     return r_regression(
-        regular_values.reshape((-1, 1)).to_list(), interpolated_values.reshape((-1, 1)).to_list()
+        regular_values.reshape((-1, 1)).to_list(),
+        interpolated_values.reshape((-1, 1)).to_list(),
     ).item()
 
 
@@ -62,7 +63,10 @@ def visualize(parameter: str, unit: str, regular_df: pl.DataFrame, interpolated_
     plt.figure(figsize=(factor * 19.2, factor * 10.8))
     plt.plot(regular_df.get_column("date"), regular_df.get_column("value"), color="red", label="regular")
     plt.plot(
-        interpolated_df.get_column("date"), interpolated_df.get_column("value"), color="black", label="interpolated"
+        interpolated_df.get_column("date"),
+        interpolated_df.get_column("value"),
+        color="black",
+        label="interpolated",
     )
     ylabel = f"{parameter.lower()} [{unit}]"
     plt.ylabel(ylabel)

diff --git a/benchmarks/interpolation_precipitation_difference.py b/benchmarks/interpolation_precipitation_difference.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 from datetime import datetime

diff --git a/benchmarks/summary_over_time.py b/benchmarks/summary_over_time.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 import os
@@ -48,7 +47,7 @@ def main():
     summarized_df = summarized_df.with_columns(
         pl.col("taken_station_id")
         .replace({"01050": "yellow", "01048": "green", "01051": "blue", "05282": "violet"})
-        .alias("color")
+        .alias("color"),
     )
 
     regular_df_01050 = get_regular_df(start_date, end_date, "01050")

diff --git a/examples/dwd_climate_summary_xarray_dump.py b/examples/dwd_climate_summary_xarray_dump.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 import os

diff --git a/examples/dwd_describe_fields.py b/examples/dwd_describe_fields.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """
@@ -37,7 +36,7 @@ def fields_example():
             resolution=DwdObservationResolution.DAILY,
             period=DwdObservationPeriod.RECENT,
             language="en",
-        )
+        ),
     )
 
     pprint(
@@ -46,7 +45,7 @@ def fields_example():
             resolution=DwdObservationResolution.DAILY,
             period=DwdObservationPeriod.RECENT,
             language="de",
-        )
+        ),
     )
 
 

diff --git a/examples/dwd_road_weather.py b/examples/dwd_road_weather.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """
@@ -29,14 +28,19 @@ def dwd_road_weather_example():
     end_date = dt.datetime.now(ZoneInfo("UTC")).replace(tzinfo=None)
     start_date = end_date - dt.timedelta(days=1)
     drw_request = DwdRoadRequest(
-        parameter="airTemperature", start_date=start_date, end_date=end_date
+        parameter="airTemperature",
+        start_date=start_date,
+        end_date=end_date,
     ).filter_by_station_id("A006")
     print(drw_request)
     df_drw = drw_request.values.all().df.drop_nulls(subset="value")
     print(df_drw)
 
     dobs_request = DwdObservationRequest(
-        parameter="temperature_air_mean_200", resolution="10_minutes", start_date=start_date, end_date=end_date
+        parameter="temperature_air_mean_200",
+        resolution="10_minutes",
+        start_date=start_date,
+        end_date=end_date,
     ).summarize(latlon=(54.8892, 8.9087))
     print(dobs_request.stations)
     df_dobs = dobs_request.df.drop_nulls(subset="value")

diff --git a/examples/mosmix_forecasts.py b/examples/mosmix_forecasts.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/observations_sql.py b/examples/observations_sql.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/observations_station_gaussian_model.py b/examples/observations_station_gaussian_model.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """
@@ -14,24 +13,29 @@
 
 """  # Noqa:D205,D400
 
+from __future__ import annotations
+
 import logging
 import os
 from pathlib import Path
-from typing import Tuple
+from typing import TYPE_CHECKING
 
 import matplotlib.pyplot as plt
 import polars as pl
-from lmfit import Parameters
-from lmfit.model import ModelResult
 
 from wetterdienst import Settings
-from wetterdienst.core.timeseries.result import StationsResult
 from wetterdienst.provider.dwd.observation import (
     DwdObservationParameter,
     DwdObservationRequest,
     DwdObservationResolution,
 )
 
+if TYPE_CHECKING:
+    from lmfit import Parameters
+    from lmfit.model import ModelResult
+
+    from wetterdienst.core.timeseries.result import StationsResult
+
 HERE = Path(__file__).parent
 
 log = logging.getLogger()
@@ -103,7 +107,7 @@ def validate_yearly_data(df: pl.DataFrame) -> bool:
             return False
         return True
 
-    def make_composite_yearly_model(self, valid_data: pl.DataFrame) -> Tuple[GaussianModel, Parameters]:
+    def make_composite_yearly_model(self, valid_data: pl.DataFrame) -> tuple[GaussianModel, Parameters]:
         """makes a composite model
         https://lmfit.github.io/lmfit-py/model.html#composite-models-adding-or-multiplying-models"""
         number_of_years = valid_data.get_column("date").dt.year().n_unique()
@@ -129,7 +133,11 @@ def make_composite_yearly_model(self, valid_data: pl.DataFrame) -> Tuple[Gaussia
 
     @staticmethod
     def model_pars_update(
-        year: int, group: pl.DataFrame, pars: Parameters, index_per_year: float, y_max: float
+        year: int,
+        group: pl.DataFrame,
+        pars: Parameters,
+        index_per_year: float,
+        y_max: float,
     ) -> Parameters:
         """updates the initial values of the model parameters"""
         idx = group.get_column("rc").to_numpy()
@@ -150,7 +158,7 @@ def plot_data_and_model(self, valid_data: pl.DataFrame, out: ModelResult, savefi
                 "year": valid_data.get_column("date"),
                 "value": valid_data.get_column("value").to_numpy(),
                 "model": out.best_fit,
-            }
+            },
         )
         title = valid_data.get_column("parameter").unique()[0]
         df.to_pandas().plot(x="year", y=["value", "model"], title=title)

diff --git a/examples/observations_stations.py b/examples/observations_stations.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/plot_german_weather_stations.py b/examples/plot_german_weather_stations.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (c) 2018-2023 earthobservations
 import os
 from pathlib import Path

diff --git a/examples/plot_hohenpeissenberg_warming_stripes.py b/examples/plot_hohenpeissenberg_warming_stripes.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (c) 2018-2023 earthobservations
 import os
 from pathlib import Path
@@ -30,7 +29,7 @@ def plot_hohenpeissenberg_warming_stripes():
     Source: https://matplotlib.org/matplotblog/posts/warming-stripes/
     """
     request = DwdObservationRequest("temperature_air_mean_200", "annual", "historical").filter_by_name(
-        "Hohenpeissenberg"
+        "Hohenpeissenberg",
     )
     df_values = request.values.all().df
     # definition of years
@@ -65,7 +64,7 @@ def plot_hohenpeissenberg_warming_stripes():
             "#cb181d",
             "#a50f15",
             "#67000d",
-        ]
+        ],
     )
     ax.set_axis_off()
     col = PatchCollection([Rectangle((y, 0), 1, 1) for y in range(first_year, last_year + 1)])

diff --git a/examples/plot_temperature_timeseries.py b/examples/plot_temperature_timeseries.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (c) 2018-2023 earthobservations
 import os
 from pathlib import Path
@@ -21,12 +20,15 @@
 def plot_temperature_timeseries():
     """Create plot for README sketch"""
     stations = DwdObservationRequest(
-        parameter="temperature_air_mean_200", resolution="daily", period="historical"
+        parameter="temperature_air_mean_200",
+        resolution="daily",
+        period="historical",
     ).filter_by_name("Hohenpeissenberg")
     df = stations.values.all().df
     df_annual = df.group_by([pl.col("date").dt.year()], maintain_order=True).agg(pl.col("value").mean().alias("value"))
     df_annual = df_annual.with_columns(
-        pl.col("date").cast(str).str.to_datetime("%Y"), pl.col("value").mean().alias("mean")
+        pl.col("date").cast(str).str.to_datetime("%Y"),
+        pl.col("value").mean().alias("mean"),
     )
     fig, ax = plt.subplots(tight_layout=True)
     df.to_pandas().plot("date", "value", ax=ax, color="blue", label="Tmean,daily", legend=False)

diff --git a/examples/radar/radar_composite_rw.py b/examples/radar/radar_composite_rw.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/radar/radar_radolan_cdc.py b/examples/radar/radar_radolan_cdc.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/radar/radar_radolan_rw.py b/examples/radar/radar_radolan_rw.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/radar/radar_scan_precip.py b/examples/radar/radar_scan_precip.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/radar/radar_scan_volume.py b/examples/radar/radar_scan_volume.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/radar/radar_site_dx.py b/examples/radar/radar_site_dx.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2021, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/radar/radar_sweep_hdf5.py b/examples/radar/radar_sweep_hdf5.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023, earthobservations developers.
 # Distributed under the MIT License. See LICENSE for more info.
 """

diff --git a/examples/wetterdienst_notebook.ipynb b/examples/wetterdienst_notebook.ipynb
@@ -681,7 +681,9 @@
     "]\n",
     "values = (\n",
     "    DwdObservationRequest(\n",
-    "        parameter=parameters, resolution=DwdObservationResolution.DAILY, period=DwdObservationPeriod.HISTORICAL\n",
+    "        parameter=parameters,\n",
+    "        resolution=DwdObservationResolution.DAILY,\n",
+    "        period=DwdObservationPeriod.HISTORICAL,\n",
     "    )\n",
     "    .filter_by_station_id(station_id=(1048,))\n",
     "    .values.all()\n",
@@ -762,7 +764,7 @@
     "for (parameter,), group in values.df.groupby([pl.col(\"parameter\")], maintain_order=True):\n",
     "    if parameter == \"precipitation_height\":\n",
     "        agg_df = group.groupby(pl.col(\"date\").dt.year(), maintain_order=True).agg(\n",
-    "            pl.when(pl.col(\"value\").is_not_null().sum() > 330).then(pl.col(\"value\").sum())\n",
+    "            pl.when(pl.col(\"value\").is_not_null().sum() > 330).then(pl.col(\"value\").sum()),\n",
     "        )\n",
     "    else:\n",
     "        agg_df = group.groupby(pl.col(\"date\").dt.year(), maintain_order=True).agg([pl.col(\"value\").mean()])\n",
@@ -835,7 +837,14 @@
     "        ax2 = ax\n",
     "    daily.to_pandas().plot(x=\"date\", y=\"value\", label=parameter, alpha=0.75, ax=ax, c=color, legend=False)\n",
     "    annual.to_pandas().plot(\n",
-    "        x=\"date\", y=\"value\", kind=\"line\", label=f\"annual({parameter})\", alpha=0.75, ax=ax2, c=\"black\", legend=False\n",
+    "        x=\"date\",\n",
+    "        y=\"value\",\n",
+    "        kind=\"line\",\n",
+    "        label=f\"annual({parameter})\",\n",
+    "        alpha=0.75,\n",
+    "        ax=ax2,\n",
+    "        c=\"black\",\n",
+    "        legend=False,\n",
     "    )\n",
     "    ax.legend(loc=0)\n",
     "    if ax != ax2:\n",