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

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

metnet/layers/ConditionWithTimeMetNet2.py

@@ -1,4 +1,5 @@
 """Condition with time how MetNet-22 does it, with FiLM layers"""
+
 import einops
 import torch
 from torch import nn as nn

metnet/layers/ConvLSTM.py

@@ -1,4 +1,5 @@
 """Originally adapted from https://github.com/aserdega/convlstmgru, MIT License Andriy Serdega"""
+
 from typing import Any, List, Optional
 
 import torch

metnet/layers/DilatedCondConv.py

@@ -1,4 +1,5 @@
 """Dilated Time Conditioned Residual Convolution Block for MetNet-2"""
+
 import torch
 import torch.nn as nn
 import torch.nn.functional as F

metnet/layers/MBConv.py

@@ -1,6 +1,7 @@
 """
 MBConv Implementation
 """
+
 from typing import Type
 
 import torch

metnet/layers/MaxViT.py

@@ -1,6 +1,7 @@
 """
 Implementation of MaxViT module
 """
+
 from dataclasses import dataclass
 from typing import List, Tuple, Type, Union
 

metnet/layers/MultiheadSelfAttention2D.py

@@ -1,6 +1,7 @@
 """
 MultiHeaded 2D Self Attention  Implementation
 """
+
 from typing import Type
 
 import torch

metnet/layers/PartitionAttention.py

@@ -1,6 +1,7 @@
 """
 Implementation of Partition (Grid and Block) Attention
 """
+
 from typing import Tuple, Type
 
 import torch

metnet/layers/RelativePositionBias.py

@@ -1,6 +1,7 @@
 """
 Implementation of Relative Position Bias
 """
+
 from typing import Tuple
 
 import torch

metnet/layers/SqueezeExcitation.py

@@ -1,6 +1,7 @@
 """
 Squeeze-and-Excitation net implementation
 """
+
 from typing import Type
 
 import torch

metnet/layers/StochasticDepth.py

@@ -1,6 +1,7 @@
 """
 Implementation of Stochastic Depth
 """
+
 import torch
 from torch import nn
 

metnet/models/metnet2.py

@@ -1,4 +1,5 @@
 """MetNet-2 model for weather forecasting"""
+
 from typing import List
 
 import torch

metnet/utils.py

@@ -1,4 +1,5 @@
 """Utilities"""
+
 import numpy as np