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