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

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

pvnet/__init__.py

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

pvnet/callbacks.py

@@ -1,5 +1,6 @@
 """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,2 +1,3 @@
 """Data parts"""
+
 from .utils import BatchSplitter

pvnet/data/base.py

@@ -1,4 +1,5 @@
 """ Data module for pytorch lightning """
+
 from datetime import datetime
 
 from lightning.pytorch import LightningDataModule

pvnet/data/pv_site_datamodule.py

@@ -1,4 +1,5 @@
 """ Data module for pytorch lightning """
+
 import glob
 
 from ocf_datapipes.batch import BatchKey, batch_to_tensor, stack_np_examples_into_batch

pvnet/data/utils.py

@@ -1,4 +1,5 @@
 """Utils common between Wind and PV datamodules"""
+
 from ocf_datapipes.batch import BatchKey, unstack_np_batch_into_examples
 from torch.utils.data import IterDataPipe, functional_datapipe
 

pvnet/data/wind_datamodule.py

@@ -1,4 +1,5 @@
 """ Data module for pytorch lightning """
+
 import glob
 
 from ocf_datapipes.batch import BatchKey, batch_to_tensor, stack_np_examples_into_batch

pvnet/load_model.py

@@ -1,4 +1,5 @@
 """ 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/single_value.py

@@ -1,4 +1,5 @@
 """Average value model"""
+
 import torch
 from ocf_datapipes.batch import BatchKey
 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,6 @@
 """Encoder modules for the satellite/NWP data based on 3D concolutions.
 """
+
 from typing import List, Union
 
 import torch

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 ocf_datapipes.batch import BatchKey, NWPBatchKey
 from torchvision.transforms.functional import center_crop
 

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/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

scripts/save_batches.py

@@ -20,6 +20,7 @@
 ```
 if wanting to override these values for example
 """
+
 # This is needed to get multiprocessing/multiple workers to behave
 try:
     import torch.multiprocessing as mp

scripts/save_concurrent_batches.py

@@ -15,6 +15,7 @@
 ```
 
 """
+
 # This is needed to get multiprocessing/multiple workers to behave
 try:
     import torch.multiprocessing as mp