From d31c0176cfdbe05aa13bdcdace9a2214715d0c17 Mon Sep 17 00:00:00 2001 From: leufen1 <l.leufen@fz-juelich.de> Date: Tue, 18 May 2021 16:25:41 +0200 Subject: [PATCH] solved lazy error, some station removals have to be fixed still --- mlair/data_handler/data_handler_mixed_sampling.py | 2 +- mlair/helpers/filter.py | 14 +++++++------- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/mlair/data_handler/data_handler_mixed_sampling.py b/mlair/data_handler/data_handler_mixed_sampling.py index 565a50df..03f10eb8 100644 --- a/mlair/data_handler/data_handler_mixed_sampling.py +++ b/mlair/data_handler/data_handler_mixed_sampling.py @@ -236,7 +236,7 @@ class DataHandlerMixedSamplingWithClimateFirFilterSingleStation(DataHandlerMixed def _extract_lazy(self, lazy_data): _data, _meta, _input_data, _target_data, self.climate_filter_coeff, self.apriori, self.all_apriori = lazy_data - DataHandlerSingleStation._extract_lazy(self, (_data, _meta, _input_data, _target_data)) + DataHandlerMixedSamplingWithFilterSingleStation._extract_lazy(self, (_data, _meta, _input_data, _target_data)) @staticmethod def _get_fs(**kwargs): diff --git a/mlair/helpers/filter.py b/mlair/helpers/filter.py index ff3bb6a5..9f7b5a6e 100644 --- a/mlair/helpers/filter.py +++ b/mlair/helpers/filter.py @@ -397,13 +397,13 @@ class ClimateFIRFilter: # filt = xr.apply_ufunc(fir_filter_vectorized, filter_input_data, time_axis, # input_core_dims=[[new_dim], []], output_core_dims=[[new_dim]], vectorize=True, # kwargs=kwargs) - # with TimeTracking(name="convolve"): - slicer = slice(int(-(length - 1) / 2), int((length - 1) / 2)) - filt = xr.apply_ufunc(fir_filter_convolve_vectorized, filter_input_data.sel({new_dim: slicer}), - input_core_dims=[[new_dim]], - output_core_dims=[[new_dim]], - vectorize=True, - kwargs={"h": h}) + with TimeTracking(name="convolve"): + slicer = slice(int(-(length - 1) / 2), int((length - 1) / 2)) + filt = xr.apply_ufunc(fir_filter_convolve_vectorized, filter_input_data.sel({new_dim: slicer}), + input_core_dims=[[new_dim]], + output_core_dims=[[new_dim]], + vectorize=True, + kwargs={"h": h}) # plot if self.plot_path is not None: -- GitLab