Fuix channel selection

openclimatefix · Apr 2, 2024 · 3c91119 · 3c91119
1 parent cd619a4
commit 3c91119
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ocf_datapipes/load/nwp/providers/excarta.py b/ocf_datapipes/load/nwp/providers/excarta.py
@@ -62,7 +62,7 @@ def open_excarta(zarr_path) -> xr.Dataset:
     nwp = nwp.drop_vars(
         ["10m_wind_speed", "10m_wind_speed_angle", "100m_wind_speed", "100m_wind_speed_angle"]
     )
-    nwp = nwp.to_dataarray(dim="channels")
+    nwp = nwp.to_dataarray(dim="channel")
     # Sanity checks.
     time = pd.DatetimeIndex(nwp.init_time_utc)
     assert time.is_unique

diff --git a/ocf_datapipes/select/filter_channels.py b/ocf_datapipes/select/filter_channels.py
@@ -32,7 +32,7 @@ def __init__(
 
     def __iter__(self) -> Union[xr.DataArray, xr.Dataset]:
         for xr_data in self.source_datapipe:
-            if "channels" not in xr_data.dims and isinstance(
+            if "channel" not in xr_data.dims and isinstance(
                 xr_data, xr.Dataset
             ):  # Variables are in their own data variables, not channels
                 logger.debug(