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

for more information, see https://pre-commit.ci
openclimatefix · Jan 29, 2025 · 9527ab4 · 9527ab4
1 parent 40ecaf4
commit 9527ab4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 9 deletions.
diff --git a/pvnet/data/site_datamodule.py b/pvnet/data/site_datamodule.py
@@ -1,12 +1,10 @@
 """ Data module for pytorch lightning """
 from glob import glob
 
-import xarray as xr
+from ocf_data_sampler.sample.site import SiteSample
 from ocf_data_sampler.torch_datasets.datasets.site import (
     SitesDataset,
-    convert_netcdf_to_numpy_sample,
 )
-from ocf_data_sampler.sample.site import SiteSample
 from torch.utils.data import Dataset
 
 from pvnet.data.base_datamodule import BaseDataModule
@@ -30,7 +28,7 @@ def __len__(self):
         return len(self.sample_paths)
 
     def __getitem__(self, idx):
-        sample = SiteSample.load(self.sample_paths[idx])        
+        sample = SiteSample.load(self.sample_paths[idx])
         return sample.to_numpy()
 
 

diff --git a/pvnet/data/uk_regional_datamodule.py b/pvnet/data/uk_regional_datamodule.py
@@ -1,9 +1,8 @@
 """ Data module for pytorch lightning """
 from glob import glob
 
-import torch
+from ocf_data_sampler.sample.uk_regional import UKRegionalSample
 from ocf_data_sampler.torch_datasets.datasets.pvnet_uk_regional import PVNetUKRegionalDataset
-from ocf_data_sampler.sample.uk_regional import UKRegionalSample 
 from torch.utils.data import Dataset
 
 from pvnet.data.base_datamodule import BaseDataModule
@@ -24,7 +23,7 @@ def __len__(self):
         return len(self.sample_paths)
 
     def __getitem__(self, idx):
-        sample = UKRegionalSample.load(self.sample_paths[idx])        
+        sample = UKRegionalSample.load(self.sample_paths[idx])
         return sample.to_numpy()
 
 

diff --git a/scripts/save_samples.py b/scripts/save_samples.py
@@ -40,10 +40,9 @@
 import shutil
 import sys
 import warnings
+
 import dask
 import hydra
-import torch
-
 from ocf_data_sampler.sample.site import SiteSample
 from ocf_data_sampler.sample.uk_regional import UKRegionalSample
 from ocf_data_sampler.torch_datasets.datasets import PVNetUKRegionalDataset, SitesDataset
@@ -67,6 +66,7 @@
 
 # -------------------------------------------------
 
+
 class SaveFuncFactory:
     """Factory for creating a function to save a sample to disk."""