formatting

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