From 552a1ae3db5045c58639a76eafa16044d63f16e0 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Tue, 27 Sep 2022 10:05:09 +0200
Subject: [PATCH] add abstract transform module

---
 mu_map/dataset/transform.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 mu_map/dataset/transform.py

diff --git a/mu_map/dataset/transform.py b/mu_map/dataset/transform.py
new file mode 100644
index 0000000..c27aaa5
--- /dev/null
+++ b/mu_map/dataset/transform.py
@@ -0,0 +1,23 @@
+from typing import List, Tuple
+
+
+from torch import Tensor
+
+
+class Transform:
+    def __call__(
+        self, inputs: Tensor, outputs_expected: Tensor
+    ) -> Tuple[Tensor, Tensor]:
+        return inputs, outputs_expected
+
+
+class SequenceTransform(Transform):
+    def __init__(self, transforms: List[Transform]):
+        self.transforms = transforms
+
+    def __call__(
+        self, inputs: Tensor, outputs_expected: Tensor
+    ) -> Tuple[Tensor, Tensor]:
+        for transforms in self.transforms:
+            inputs, outputs_expected = transforms(inputs, outputs_expected)
+        return inputs, outputs_expected
-- 
GitLab