diff --git a/pvnet/__init__.py b/pvnet/__init__.py index bf59ae4f..2a92797d 100644 --- a/pvnet/__init__.py +++ b/pvnet/__init__.py @@ -1,2 +1,3 @@ """PVNet""" + __version__ = "3.0.5" diff --git a/pvnet/callbacks.py b/pvnet/callbacks.py index 0455d774..14cc15bf 100644 --- a/pvnet/callbacks.py +++ b/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 diff --git a/pvnet/data/__init__.py b/pvnet/data/__init__.py index 5d763df0..7f3aa89f 100644 --- a/pvnet/data/__init__.py +++ b/pvnet/data/__init__.py @@ -1,2 +1,3 @@ """Data parts""" + from .utils import BatchSplitter diff --git a/pvnet/data/base.py b/pvnet/data/base.py index b2227f7d..dea8d08d 100644 --- a/pvnet/data/base.py +++ b/pvnet/data/base.py @@ -1,4 +1,5 @@ """ Data module for pytorch lightning """ + from datetime import datetime import torch diff --git a/pvnet/data/pv_site_datamodule.py b/pvnet/data/pv_site_datamodule.py index 7d49f8bd..ba6dfa7a 100644 --- a/pvnet/data/pv_site_datamodule.py +++ b/pvnet/data/pv_site_datamodule.py @@ -1,4 +1,5 @@ """ Data module for pytorch lightning """ + import glob from ocf_datapipes.batch import BatchKey, stack_np_examples_into_batch diff --git a/pvnet/data/utils.py b/pvnet/data/utils.py index 6100a23c..f0ca9c9a 100644 --- a/pvnet/data/utils.py +++ b/pvnet/data/utils.py @@ -1,4 +1,5 @@ """Utils common between Wind and PV datamodules""" + import numpy as np import torch from ocf_datapipes.batch import BatchKey, unstack_np_batch_into_examples diff --git a/pvnet/data/wind_datamodule.py b/pvnet/data/wind_datamodule.py index eec3365b..94bd85ba 100644 --- a/pvnet/data/wind_datamodule.py +++ b/pvnet/data/wind_datamodule.py @@ -1,4 +1,5 @@ """ Data module for pytorch lightning """ + import glob from ocf_datapipes.batch import BatchKey, stack_np_examples_into_batch diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py index db56946f..e3bc3b79 100644 --- a/pvnet/models/base_model.py +++ b/pvnet/models/base_model.py @@ -1,4 +1,5 @@ """Base model for all PVNet submodels""" + import json import logging import os diff --git a/pvnet/models/baseline/single_value.py b/pvnet/models/baseline/single_value.py index 980feb95..44ac2061 100644 --- a/pvnet/models/baseline/single_value.py +++ b/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 diff --git a/pvnet/models/multimodal/encoders/basic_blocks.py b/pvnet/models/multimodal/encoders/basic_blocks.py index 14cc610f..d65b4bd0 100644 --- a/pvnet/models/multimodal/encoders/basic_blocks.py +++ b/pvnet/models/multimodal/encoders/basic_blocks.py @@ -1,4 +1,5 @@ """Basic blocks for image sequence encoders""" + from abc import ABCMeta, abstractmethod import torch diff --git a/pvnet/models/multimodal/encoders/encoders3d.py b/pvnet/models/multimodal/encoders/encoders3d.py index 9ff6672a..80e9b46c 100644 --- a/pvnet/models/multimodal/encoders/encoders3d.py +++ b/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 diff --git a/pvnet/models/multimodal/linear_networks/basic_blocks.py b/pvnet/models/multimodal/linear_networks/basic_blocks.py index c115c9fb..084b585a 100644 --- a/pvnet/models/multimodal/linear_networks/basic_blocks.py +++ b/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 diff --git a/pvnet/models/multimodal/linear_networks/networks.py b/pvnet/models/multimodal/linear_networks/networks.py index 0009afa0..b52c0201 100644 --- a/pvnet/models/multimodal/linear_networks/networks.py +++ b/pvnet/models/multimodal/linear_networks/networks.py @@ -1,4 +1,5 @@ """Linear networks used for the fusion model""" + from torch import nn from pvnet.models.multimodal.linear_networks.basic_blocks import ( diff --git a/pvnet/models/multimodal/nwp_weighting.py b/pvnet/models/multimodal/nwp_weighting.py index a34c222b..8fe21ff8 100644 --- a/pvnet/models/multimodal/nwp_weighting.py +++ b/pvnet/models/multimodal/nwp_weighting.py @@ -1,4 +1,5 @@ """Architecture for simple learned weighted average of the downwards short wave radiation flux""" + from typing import Optional import torch diff --git a/pvnet/models/multimodal/site_encoders/basic_blocks.py b/pvnet/models/multimodal/site_encoders/basic_blocks.py index b20835f1..66f4fb46 100644 --- a/pvnet/models/multimodal/site_encoders/basic_blocks.py +++ b/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 diff --git a/pvnet/models/multimodal/weather_residual.py b/pvnet/models/multimodal/weather_residual.py index f4bf36b9..e7f511fb 100644 --- a/pvnet/models/multimodal/weather_residual.py +++ b/pvnet/models/multimodal/weather_residual.py @@ -1,4 +1,5 @@ """Model architecture where preidction is split into climateology and weather residual""" + from collections import OrderedDict from typing import Optional diff --git a/pvnet/training.py b/pvnet/training.py index cc5f3c43..0b2339b9 100644 --- a/pvnet/training.py +++ b/pvnet/training.py @@ -1,4 +1,5 @@ """Training""" + import os import shutil from typing import Optional diff --git a/pvnet/utils.py b/pvnet/utils.py index ccfa6826..493e62e3 100644 --- a/pvnet/utils.py +++ b/pvnet/utils.py @@ -1,4 +1,5 @@ """Utils""" + import logging import os import warnings diff --git a/scripts/checkpoint_to_huggingface.py b/scripts/checkpoint_to_huggingface.py index f5777114..cb3ed7a2 100644 --- a/scripts/checkpoint_to_huggingface.py +++ b/scripts/checkpoint_to_huggingface.py @@ -5,6 +5,7 @@ --local-path="~/tmp/this_model" \ --no-push-to-hub """ + import glob import os import tempfile diff --git a/scripts/save_batches.py b/scripts/save_batches.py index c8983966..f7abb21f 100644 --- a/scripts/save_batches.py +++ b/scripts/save_batches.py @@ -15,6 +15,7 @@ ``` """ + import logging import os import shutil diff --git a/setup.py b/setup.py index dd15b3a5..eaa045e9 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,5 @@ """ Usual setup file for package """ + # read the contents of your README file from pathlib import Path