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

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

pseudo_labeller/model/__init__.py

@@ -1,2 +1,3 @@
 """Models"""
+
 from pseudo_labeller.model.idam import PsuedoIrradienceForecastor

pseudo_labeller/model/idam.py

@@ -1,4 +1,5 @@
 """Psuedo-irradience forecastor/labeller"""
+
 import einops
 import torch
 import torch.nn as nn
@@ -68,7 +69,7 @@ def __init__(
                 in_channels=input_channels,
                 out_channels=conv3d_channels,
                 kernel_size=(kernel_size, kernel_size, kernel_size),
-                padding=(1,0,0),
+                padding=(1, 0, 0),
             )
         )
         for i in range(0, num_layers):
@@ -77,7 +78,7 @@ def __init__(
                     in_channels=conv3d_channels,
                     out_channels=conv3d_channels,
                     kernel_size=(kernel_size, kernel_size, kernel_size),
-                    padding=(1,0,0),
+                    padding=(1, 0, 0),
                 )
             )
 
@@ -95,9 +96,7 @@ def __init__(
         # Small head model to convert from latent space to PV generation for training
         # Input is per-pixel input data, this will be
         # reshaped to the same output steps as the latent head
-        self.pv_meta_input = nn.Linear(
-            pv_meta_input_channels, out_features=hidden_dim
-        )
+        self.pv_meta_input = nn.Linear(pv_meta_input_channels, out_features=hidden_dim)
 
         # Output is forecast steps channels, each channel is a timestep
         # For labelling, this should be 1, forecasting the middle
@@ -142,7 +141,5 @@ def forward(self, x: torch.Tensor, pv_meta: torch.Tensor = None, output_latents:
         x = torch.cat([x, pv_meta], dim=1)
         # Get pv_meta_output
         x = self.pv_meta_output(x)
-        x = F.relu(
-            self.pv_meta_output2(x)
-        )  # Generation can only be positive or 0, so ReLU
+        x = F.relu(self.pv_meta_output2(x))  # Generation can only be positive or 0, so ReLU
         return x

setup.py

@@ -1,4 +1,5 @@
 """ Usual setup file for package """
+
 # read the contents of your README file
 from pathlib import Path