From 85930b5e113e4dbbcf0e66cb49d6ff2587f6eb3a Mon Sep 17 00:00:00 2001 From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de> Date: Tue, 27 Sep 2022 10:05:29 +0200 Subject: [PATCH] formatting --- mu_map/dataset/normalization.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/mu_map/dataset/normalization.py b/mu_map/dataset/normalization.py index 85ec9af..f233b4c 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 -- GitLab