diff --git a/mlair/data_handler/data_handler_mixed_sampling.py b/mlair/data_handler/data_handler_mixed_sampling.py index 0a7adadb7d3c00ebd10dbff176feb4a8ff6b8d5f..f49cee73c0736384bc878751c3c26e968af91147 100644 --- a/mlair/data_handler/data_handler_mixed_sampling.py +++ b/mlair/data_handler/data_handler_mixed_sampling.py @@ -75,10 +75,8 @@ class DataHandlerMixedSamplingSingleStation(DataHandlerSingleStation): return data def set_inputs_and_targets(self): - inputs = self._data[0].sel({self.target_dim: helpers.to_list(self.variables)}) - targets = self._data[1].sel({self.target_dim: self.target_var}) - self.input_data.data = inputs - self.target_data.data = targets + self.input_data = self._data[0].sel({self.target_dim: helpers.to_list(self.variables)}) + self.target_data = self._data[1].sel({self.target_dim: helpers.to_list(self.target_var)}) def setup_data_path(self, data_path, sampling): """Sets two paths instead of single path. Expects sampling arg to be a list with two entries"""