Fuix channel selection

Author: jacobbieker

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

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(