[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

nwp/excarta/merge_excarta.py

@@ -4,13 +4,10 @@
 import xarray as xr
 
 
-
 def merge_zarr_files(zarr_path, merged_zarr_path):
     # Collect paths of Zarr files in the specified directory
     zarr_files = [
-        os.path.join(zarr_path, file)
-        for file in os.listdir(zarr_path)
-        if file.endswith(".zarr")
+        os.path.join(zarr_path, file) for file in os.listdir(zarr_path) if file.endswith(".zarr")
     ]
 
     print("1")

nwp/excarta/parse_excarta_monthly.py

@@ -7,7 +7,6 @@
 import xarray as xr
 
 
-
 def _parse_args():
     parser = argparse.ArgumentParser()
     parser.add_argument("output", type=pathlib.Path, help="Output zarr file")
@@ -81,8 +80,7 @@ def pdtocdf(datasets):
     """
 
     datasets = [
-        ds.set_index(index=["init_time", "step", "Latitude", "Longitude"])
-        for ds in datasets
+        ds.set_index(index=["init_time", "step", "Latitude", "Longitude"]) for ds in datasets
     ]
 
     ds = xr.concat(datasets, dim="index")

nwp/excarta/parse_excarta_to_output.py

@@ -71,9 +71,7 @@ def pdtocdf(dfs):
     merged_df = pd.concat(dfs, ignore_index=True)
 
     ds = xr.Dataset.from_dataframe(merged_df)
-    ds = ds.set_index(index=["init_time", "step", "Latitude", "Longitude"]).unstack(
-        "index"
-    )
+    ds = ds.set_index(index=["init_time", "step", "Latitude", "Longitude"]).unstack("index")
     ds = ds.drop_vars(["LocationId", "DateTimeUTC"])
 
     var_names = ds.data_vars

nwp/excarta/parse_excarta_to_output_low_mem.py

@@ -70,8 +70,7 @@ def pdtocdf(datasets):
     #     ds = xr.merge(datasets)
 
     datasets = [
-        ds.set_index(index=["init_time", "step", "Latitude", "Longitude"])
-        for ds in datasets
+        ds.set_index(index=["init_time", "step", "Latitude", "Longitude"]) for ds in datasets
     ]
 
     ds = xr.concat(datasets, dim="index")

nwp/icon/app.py

@@ -81,20 +81,12 @@ def process_model_files(
         var_3d_list = GLOBAL_VAR3D_LIST
         var_2d_list = GLOBAL_VAR2D_LIST
         lon_ds = xr.open_dataset(
-            list(
-                glob(
-                    os.path.join(folder, run, f"{var_base}_time-invariant_*_CLON.grib2")
-                )
-            )[0],
+            list(glob(os.path.join(folder, run, f"{var_base}_time-invariant_*_CLON.grib2")))[0],
             engine="cfgrib",
             backend_kwargs={"errors": "ignore"},
         )
         lat_ds = xr.open_dataset(
-            list(
-                glob(
-                    os.path.join(folder, run, f"{var_base}_time-invariant_*_CLAT.grib2")
-                )
-            )[0],
+            list(glob(os.path.join(folder, run, f"{var_base}_time-invariant_*_CLAT.grib2")))[0],
             engine="cfgrib",
             backend_kwargs={"errors": "ignore"},
         )
@@ -156,9 +148,7 @@ def process_model_files(
         print(var_2d)
         try:
             ds = xr.open_mfdataset(
-                os.path.join(
-                    folder, run, f"{var_base}_single-level_*_*_{var_2d.upper()}.grib2"
-                ),
+                os.path.join(folder, run, f"{var_base}_single-level_*_*_{var_2d.upper()}.grib2"),
                 engine="cfgrib",
                 combine="nested",
                 concat_dim="step",
@@ -203,9 +193,7 @@ def upload_to_hf(dataset_xr, folder, model="global", run="00", token=None):
             "longitude": 350,
             "isobaricInhPa": -1,
         }
-    encoding = {
-        var: {"compressor": Blosc2("zstd", clevel=9)} for var in dataset_xr.data_vars
-    }
+    encoding = {var: {"compressor": Blosc2("zstd", clevel=9)} for var in dataset_xr.data_vars}
     encoding["time"] = {"units": "nanoseconds since 1970-01-01"}
     with zarr.ZipStore(
         zarr_path,