From b745005154defbe46e2a2934854315cd4069948b Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Thu, 11 Nov 2021 11:38:05 +0100
Subject: [PATCH] use dim name instead of pos to be independent of dim ordering
---
mlair/data_handler/data_handler_single_station.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py
index 6190784f..c898ca14 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))
--
GitLab