From aa858458ad326f3b24e5d88eee78fac0525b80d7 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Wed, 19 Jan 2022 14:58:25 +0100
Subject: [PATCH] also ensure variables order in get_X
---
mlair/data_handler/data_handler_single_station.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py
index 038851d0..4217583d 100644
--- a/mlair/data_handler/data_handler_single_station.py
+++ b/mlair/data_handler/data_handler_single_station.py
@@ -163,7 +163,7 @@ class DataHandlerSingleStation(AbstractDataHandler):
return self.label.squeeze([self.iter_dim, self.target_dim]).transpose(self.time_dim, self.window_dim).copy()
def get_X(self, **kwargs):
- return self.get_transposed_history()
+ return self.get_transposed_history().sel({self.target_dim: self.variables})
def get_Y(self, **kwargs):
return self.get_transposed_label()
--
GitLab