From 29fbde927d6a58ee98c9274e6475f33bba3f4c45 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 30 Mar 2021 18:22:33 +0200
Subject: [PATCH] extarct method
---
mlair/data_handler/data_handler_single_station.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py
index c5b3ae8d..cb0a4803 100644
--- a/mlair/data_handler/data_handler_single_station.py
+++ b/mlair/data_handler/data_handler_single_station.py
@@ -702,6 +702,12 @@ class DataHandlerSingleStation(AbstractDataHandler):
:param inverse:
:return:
"""
+ pos = self.get_transformation_base(base)
+ return self.transform(data, dim=dim, opts=self._transformation[pos], inverse=inverse,
+ transformation_dim=self.target_dim)
+
+ @staticmethod
+ def get_transformation_base(base):
if base in ["target", 1]:
pos = 1
elif base in ["input", 0]:
@@ -710,8 +716,7 @@ class DataHandlerSingleStation(AbstractDataHandler):
raise ValueError("apply transformation requires a reference for transformation options. Please specify if"
"you want to use input or target transformation using the parameter 'base'. Given was: " +
base)
- return self.transform(data, dim=dim, opts=self._transformation[pos], inverse=inverse,
- transformation_dim=self.target_dim)
+ return pos
def _hash_list(self):
return sorted(list(set(self._hash)))
--
GitLab