[pre-commit.ci] pre-commit autoupdate

.pre-commit-config.yaml

@@ -14,18 +14,18 @@ repos:
   # python code formatting/linting
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: "v0.1.6"
+    rev: "v0.2.2"
     hooks:
       - id: ruff
         args: [--fix]
   - repo: https://github.com/psf/black
-    rev: 23.11.0
+    rev: 24.2.0
     hooks:
       - id: black
         args: [--line-length, "100"]
   # yaml formatting
   - repo: https://github.com/pre-commit/mirrors-prettier
-    rev: v3.1.0
+    rev: v4.0.0-alpha.8
     hooks:
       - id: prettier
         types: [yaml]

graph_weather/__init__.py

@@ -1,3 +1,4 @@
 """Main import for the complete models"""
+
 from .models.analysis import GraphWeatherAssimilator
 from .models.forecast import GraphWeatherForecaster

graph_weather/models/__init__.py

@@ -1,4 +1,5 @@
 """Models"""
+
 from .layers.assimilator_decoder import AssimilatorDecoder
 from .layers.assimilator_encoder import AssimilatorEncoder
 from .layers.decoder import Decoder

graph_weather/models/analysis.py

@@ -1,4 +1,5 @@
 """Model for forecasting weather from NWP states"""
+
 import torch
 from huggingface_hub import PyTorchModelHubMixin
 

graph_weather/models/forecast.py

@@ -1,4 +1,5 @@
 """Model for forecasting weather from NWP states"""
+
 from typing import Optional
 
 import torch

graph_weather/models/layers/assimilator_decoder.py

@@ -13,6 +13,7 @@
 more complex dynamics)
 
 """
+
 import einops
 import h3
 import numpy as np

graph_weather/models/layers/assimilator_encoder.py

@@ -21,6 +21,7 @@
 
 
 """
+
 from typing import Tuple
 
 import einops

graph_weather/models/layers/decoder.py

@@ -13,6 +13,7 @@
 more complex dynamics)
 
 """
+
 import torch
 
 from graph_weather.models.layers.assimilator_decoder import AssimilatorDecoder

graph_weather/models/layers/encoder.py

@@ -21,6 +21,7 @@
 
 
 """
+
 from typing import Tuple
 
 import einops

graph_weather/models/layers/graph_net_block.py

@@ -4,6 +4,7 @@
 This code is taken from https://github.com/CCSI-Toolset/MGN which is available under the
 US Government License
 """
+
 from typing import Optional, Tuple
 
 import torch

graph_weather/models/layers/processor.py

@@ -7,6 +7,7 @@
 and its immediate neighbors. There are residual connections between each round of processing.
 
 """
+
 import torch
 
 from graph_weather.models.layers.graph_net_block import GraphProcessor

graph_weather/models/losses.py

@@ -1,4 +1,5 @@
 """Weather loss functions"""
+
 import numpy as np
 import torch
 

setup.py

@@ -1,4 +1,5 @@
 """Setup"""
+
 from pathlib import Path
 
 from setuptools import find_packages, setup

train/pl_graph_weather.py

@@ -1,4 +1,5 @@
 """PyTorch Lightning training script for the weather forecasting model"""
+
 import click
 import datasets
 import numpy as np

train/run_fulll.py

@@ -1,4 +1,5 @@
 """Training script for training the weather forecasting model"""
+
 import json
 import os
 import sys