diff --git a/mu_map/dataset/normalization.py b/mu_map/dataset/normalization.py
index 85ec9afc52c2905182c6d9705b2990706eb5a81f..f233b4c87d9468ba96c344d4494f74bcc1ecf654 100644
--- a/mu_map/dataset/normalization.py
+++ b/mu_map/dataset/normalization.py
@@ -8,7 +8,9 @@ def norm_max(tensor: Tensor) -> Tensor:
 
 
 class MaxNormTransform(Transform):
-    def __call__(self, inputs: Tensor, outputs_expected: Tensor) -> Tuple[Tensor, Tensor]:
+    def __call__(
+        self, inputs: Tensor, outputs_expected: Tensor
+    ) -> Tuple[Tensor, Tensor]:
         return norm_max(inputs), outputs_expected
 
 
@@ -17,7 +19,9 @@ def norm_mean(tensor: Tensor):
 
 
 class MeanNormTransform(Transform):
-    def __call__(self, inputs: Tensor, outputs_expected: Tensor) -> Tuple[Tensor, Tensor]:
+    def __call__(
+        self, inputs: Tensor, outputs_expected: Tensor
+    ) -> Tuple[Tensor, Tensor]:
         return norm_mean(inputs), outputs_expected
 
 
@@ -26,7 +30,9 @@ def norm_gaussian(tensor: Tensor):
 
 
 class GaussianNormTransform(Transform):
-    def __call__(self, inputs: Tensor, outputs_expected: Tensor) -> Tuple[Tensor, Tensor]:
+    def __call__(
+        self, inputs: Tensor, outputs_expected: Tensor
+    ) -> Tuple[Tensor, Tensor]:
         return norm_gaussian(inputs), outputs_expected