diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py index 6190784f27a3257c12cf7f6228d5e10324c61a89..c898ca14eee8fad51371c4889d1c2f81aacf7727 100644 --- a/mlair/data_handler/data_handler_single_station.py +++ b/mlair/data_handler/data_handler_single_station.py @@ -586,11 +586,11 @@ class DataHandlerSingleStation(AbstractDataHandler): non_nan_history = self.history.dropna(dim=dim) non_nan_label = self.label.dropna(dim=dim) non_nan_observation = self.observation.dropna(dim=dim) - if non_nan_label.shape[2] == 0: + if non_nan_label.coords[dim].shape[0] == 0: raise ValueError(f'self.label consist of NaNs only - station {self.station} is therefore dropped') - if non_nan_history.shape[2] == 0: + if non_nan_history.coords[dim].shape[0] == 0: raise ValueError(f'self.history consist of NaNs only - station {self.station} is therefore dropped') - if non_nan_observation.shape[2] == 0: + if non_nan_observation.coords[dim].shape[0] == 0: raise ValueError(f'self.observation consist of NaNs only - station {self.station} is therefore dropped') intersect = reduce(np.intersect1d, (non_nan_history.coords[dim].values, non_nan_label.coords[dim].values, non_nan_observation.coords[dim].values))