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

for more information, see https://pre-commit.ci
openclimatefix · Nov 21, 2023 · d74d7ed · d74d7ed
1 parent 3839fe8
commit d74d7ed
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 12 deletions.
diff --git a/pvnet/data/datamodule.py b/pvnet/data/datamodule.py
@@ -3,16 +3,14 @@
 
 import numpy as np
 import torch
-from torch.utils.data import DataLoader
-from torch.utils.data.datapipes.datapipe import IterDataPipe
-from torch.utils.data.datapipes.iter import FileLister
-from torch.utils.data.datapipes._decorator import functional_datapipe
-
 from lightning.pytorch import LightningDataModule
 from ocf_datapipes.training.pvnet import pvnet_datapipe
 from ocf_datapipes.utils.consts import BatchKey
 from ocf_datapipes.utils.utils import stack_np_examples_into_batch
-
+from torch.utils.data import DataLoader
+from torch.utils.data.datapipes._decorator import functional_datapipe
+from torch.utils.data.datapipes.datapipe import IterDataPipe
+from torch.utils.data.datapipes.iter import FileLister
 
 
 def copy_batch_to_device(batch, device):

diff --git a/scripts/save_batches.py b/scripts/save_batches.py
@@ -25,13 +25,11 @@
 
 import hydra
 import torch
-from torch.utils.data import DataLoader
-from torch.utils.data.datapipes.datapipe import IterDataPipe
 from ocf_datapipes.training.pvnet import pvnet_datapipe
 from ocf_datapipes.utils.utils import stack_np_examples_into_batch
 from omegaconf import DictConfig, OmegaConf
 from sqlalchemy import exc as sa_exc
-
+from torch.utils.data import DataLoader
 from tqdm import tqdm
 
 from pvnet.data.datamodule import batch_to_tensor

diff --git a/scripts/save_concurrent_batches.py b/scripts/save_concurrent_batches.py
@@ -23,8 +23,6 @@
 import hydra
 import numpy as np
 import torch
-from torch.utils.data import DataLoader
-from torch.utils.data.datapipes.datapipe import IterDataPipe
 from ocf_datapipes.training.common import (
     open_and_return_datapipes,
 )
@@ -33,7 +31,7 @@
 from ocf_datapipes.utils.utils import stack_np_examples_into_batch
 from omegaconf import DictConfig, OmegaConf
 from sqlalchemy import exc as sa_exc
-
+from torch.utils.data import DataLoader
 from tqdm import tqdm
 
 from pvnet.data.datamodule import batch_to_tensor