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

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

experiments/india/004_n_training_samples/log-plot.py

@@ -1,4 +1,4 @@
-""" Small script to make MAE vs number of batches plot"""
+"""Small script to make MAE vs number of batches plot"""
 
 import pandas as df
 import plotly.graph_objects as go

pvnet/__init__.py

@@ -1,2 +1,3 @@
 """PVNet"""
+
 __version__ = "4.0.1"

pvnet/callbacks.py

@@ -1,5 +1,5 @@
-"""Custom callbacks
-"""
+"""Custom callbacks"""
+
 from lightning.pytorch import Trainer
 from lightning.pytorch.callbacks import BaseFinetuning, EarlyStopping, LearningRateFinder
 from lightning.pytorch.trainer.states import TrainerFn

pvnet/data/__init__.py

@@ -1,3 +1,4 @@
 """Data parts"""
+
 from .site_datamodule import SiteDataModule
 from .uk_regional_datamodule import DataModule

pvnet/data/base_datamodule.py

@@ -1,4 +1,4 @@
-""" Data module for pytorch lightning """
+"""Data module for pytorch lightning"""
 
 from glob import glob
 

pvnet/data/site_datamodule.py

@@ -1,4 +1,4 @@
-""" Data module for pytorch lightning """
+"""Data module for pytorch lightning"""
 
 from ocf_data_sampler.sample.site import SiteSample
 from ocf_data_sampler.torch_datasets.datasets.site import SitesDataset

pvnet/data/uk_regional_datamodule.py

@@ -1,4 +1,4 @@
-""" Data module for pytorch lightning """
+"""Data module for pytorch lightning"""
 
 from ocf_data_sampler.sample.uk_regional import UKRegionalSample
 from ocf_data_sampler.torch_datasets.datasets.pvnet_uk import PVNetUKRegionalDataset

pvnet/load_model.py

@@ -1,4 +1,5 @@
-""" Load a model from its checkpoint directory """
+"""Load a model from its checkpoint directory"""
+
 import glob
 import os
 

pvnet/models/base_model.py

@@ -1,4 +1,5 @@
 """Base model for all PVNet submodels"""
+
 import json
 import logging
 import os

pvnet/models/baseline/last_value.py

@@ -1,6 +1,5 @@
 """Persistence model"""
 
-
 import pvnet
 from pvnet.models.base_model import BaseModel
 from pvnet.optimizers import AbstractOptimizer

pvnet/models/baseline/single_value.py

@@ -1,4 +1,5 @@
 """Average value model"""
+
 import torch
 from torch import nn
 

pvnet/models/ensemble.py

@@ -1,4 +1,5 @@
 """Model which uses mutliple prediction heads"""
+
 from typing import Optional
 
 import torch

pvnet/models/multimodal/encoders/basic_blocks.py

@@ -1,4 +1,5 @@
 """Basic blocks for image sequence encoders"""
+
 from abc import ABCMeta, abstractmethod
 
 import torch

pvnet/models/multimodal/encoders/encoders3d.py

@@ -1,5 +1,5 @@
-"""Encoder modules for the satellite/NWP data based on 3D concolutions.
-"""
+"""Encoder modules for the satellite/NWP data based on 3D concolutions."""
+
 from typing import List, Union
 
 import torch

pvnet/models/multimodal/encoders/encodersRNN.py

@@ -1,5 +1,4 @@
-"""Encoder modules for the satellite/NWP data based on recursive and 2D convolutional layers.
-"""
+"""Encoder modules for the satellite/NWP data based on recursive and 2D convolutional layers."""
 
 import torch
 from torch import nn

pvnet/models/multimodal/linear_networks/basic_blocks.py

@@ -1,4 +1,5 @@
 """Basic blocks for the lienar networks"""
+
 from abc import ABCMeta, abstractmethod
 from collections import OrderedDict
 

pvnet/models/multimodal/linear_networks/networks.py

@@ -1,4 +1,5 @@
 """Linear networks used for the fusion model"""
+
 from torch import nn, rand
 
 from pvnet.models.multimodal.linear_networks.basic_blocks import (

pvnet/models/multimodal/multimodal_base.py

@@ -1,4 +1,5 @@
 """Base model class for multimodal model and unimodal teacher"""
+
 from torchvision.transforms.functional import center_crop
 
 from pvnet.models.base_model import BaseModel

pvnet/models/multimodal/site_encoders/basic_blocks.py

@@ -1,4 +1,5 @@
 """Basic blocks for PV-site encoders"""
+
 from abc import ABCMeta, abstractmethod
 
 from torch import nn

pvnet/models/multimodal/site_encoders/encoders.py

@@ -1,6 +1,4 @@
-"""Encoder modules for the site-level PV data.
-
-"""
+"""Encoder modules for the site-level PV data."""
 
 import einops
 import torch

pvnet/optimizers.py

@@ -1,5 +1,4 @@
-"""Optimizer factory-function classes.
-"""
+"""Optimizer factory-function classes."""
 
 from abc import ABC, abstractmethod
 

pvnet/training.py

@@ -1,4 +1,5 @@
 """Training"""
+
 import os
 import shutil
 from typing import Optional

pvnet/utils.py

@@ -1,4 +1,5 @@
 """Utils"""
+
 import logging
 import warnings
 from collections.abc import Sequence

run.py

@@ -1,5 +1,4 @@
-"""Run training
-"""
+"""Run training"""
 
 import os
 

scripts/save_concurrent_samples.py

@@ -15,6 +15,7 @@
 ```
 
 """
+
 # Ensure this block of code runs only in the main process to avoid issues with worker processes.
 if __name__ == "__main__":
     import torch.multiprocessing as mp