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

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

pvnet/data/utils.py

@@ -1,8 +1,7 @@
 """Utils common between Wind and PV datamodules"""
 import numpy as np
 import torch
-from ocf_datapipes.batch import unstack_np_batch_into_examples
-from ocf_datapipes.batch import BatchKey
+from ocf_datapipes.batch import BatchKey, unstack_np_batch_into_examples
 from torch.utils.data import IterDataPipe, functional_datapipe
 
 

pvnet/utils.py

@@ -15,8 +15,8 @@
 import yaml
 from lightning.pytorch.loggers import Logger
 from lightning.pytorch.utilities import rank_zero_only
-from ocf_datapipes.utils import Location
 from ocf_datapipes.batch import BatchKey
+from ocf_datapipes.utils import Location
 from ocf_datapipes.utils.geospatial import osgb_to_lon_lat
 from omegaconf import DictConfig, OmegaConf
 

scripts/hindcast.py

@@ -10,9 +10,9 @@
 import pandas as pd
 import torch
 import xarray as xr
+from ocf_datapipes.batch import BatchKey
 from ocf_datapipes.load import OpenGSP
 from ocf_datapipes.training.pvnet import construct_sliced_data_pipeline
-from ocf_datapipes.batch import BatchKey
 from ocf_datapipes.utils.utils import stack_np_examples_into_batch
 from torchdata.dataloader2 import DataLoader2, MultiProcessingReadingService
 from torchdata.datapipes.iter import IterableWrapper

scripts/save_concurrent_batches.py

@@ -23,12 +23,11 @@
 import hydra
 import numpy as np
 import torch
-from ocf_datapipes.batch import stack_np_examples_into_batch
+from ocf_datapipes.batch import BatchKey, stack_np_examples_into_batch
 from ocf_datapipes.training.common import (
     open_and_return_datapipes,
 )
 from ocf_datapipes.training.pvnet import construct_loctime_pipelines, construct_sliced_data_pipeline
-from ocf_datapipes.batch import BatchKey
 from omegaconf import DictConfig, OmegaConf
 from sqlalchemy import exc as sa_exc
 from torch.utils.data import DataLoader