[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
openclimatefix · Nov 9, 2021 · 8af6638 · 8af6638
1 parent 8520569
commit 8af6638
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/nowcasting_gan/common.py b/nowcasting_gan/common.py
@@ -420,9 +420,11 @@ def __init__(
         self.use_attention = use_attention
         self.distribution = normal.Normal(loc=torch.Tensor([0.0]), scale=torch.Tensor([1.0]))
 
-        self.conv_3x3 = spectral_norm(torch.nn.Conv2d(
-            in_channels=shape[0], out_channels=shape[0], kernel_size=(3, 3), padding=1
-        ))
+        self.conv_3x3 = spectral_norm(
+            torch.nn.Conv2d(
+                in_channels=shape[0], out_channels=shape[0], kernel_size=(3, 3), padding=1
+            )
+        )
         self.l_block1 = LBlock(input_channels=shape[0], output_channels=output_channels // 32)
         self.l_block2 = LBlock(
             input_channels=output_channels // 32, output_channels=output_channels // 16