From 9f35efe547513a10fe8380ab50d8bb34035963dc Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Tue, 1 Dec 2020 14:46:00 +0100
Subject: [PATCH] DataHandlerMixedSampling can now differ between input and
target values and loads only required variables (not checked for all other
handlers with mixed samplings yet)
---
mlair/data_handler/data_handler_mixed_sampling.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mlair/data_handler/data_handler_mixed_sampling.py b/mlair/data_handler/data_handler_mixed_sampling.py
index 80890b6f..8aedbf28 100644
--- a/mlair/data_handler/data_handler_mixed_sampling.py
+++ b/mlair/data_handler/data_handler_mixed_sampling.py
@@ -36,7 +36,9 @@ class DataHandlerMixedSamplingSingleStation(DataHandlerSingleStation):
self.make_samples()
def load_and_interpolate(self, ind) -> [xr.DataArray, pd.DataFrame]:
- data, self.meta = self.load_data(self.path[ind], self.station, self.statistics_per_var, self.sampling[ind],
+ vars = [self.variables, self.target_var]
+ stats_per_var = helpers.select_from_dict(self.statistics_per_var, vars[ind])
+ data, self.meta = self.load_data(self.path[ind], self.station, stats_per_var, self.sampling[ind],
self.station_type, self.network, self.store_data_locally, self.data_origin,
self.start, self.end)
data = self.interpolate(data, dim=self.time_dim, method=self.interpolation_method,
--
GitLab