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

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

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