[pre-commit.ci] pre-commit autoupdate

.pre-commit-config.yaml

@@ -3,7 +3,7 @@ default_language_version:
 
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.4.0
+    rev: v5.0.0
     hooks:
       # list of supported hooks: https://pre-commit.com/hooks.html
       - id: trailing-whitespace
@@ -12,20 +12,20 @@ repos:
       - id: detect-private-key
 
   # python code formatting/linting
-  - repo: https://github.com/charliermarsh/ruff-pre-commit
+  - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: "v0.0.261"
+    rev: "v0.6.9"
     hooks:
       - id: ruff
         args: [--fix]
   - repo: https://github.com/psf/black
-    rev: 23.3.0
+    rev: 24.8.0
     hooks:
       - id: black
         args: [--line-length, "100"]
   # yaml formatting
   - repo: https://github.com/pre-commit/mirrors-prettier
-    rev: v3.0.0-alpha.6
+    rev: v4.0.0-alpha.8
     hooks:
       - id: prettier
         types: [yaml]

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