Merge remote-tracking branch 'origin/jacob/netcdf' into jacob/netcdf

openclimatefix · Nov 15, 2023 · bf1a79a · bf1a79a
2 parents 826e377 + a43f5a3
commit bf1a79a
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/ocf_datapipes/training/windnet.py b/ocf_datapipes/training/windnet.py
@@ -6,11 +6,15 @@
 import numpy as np
 import xarray as xr
 from torchdata.datapipes import functional_datapipe
-from torchdata.datapipes.iter import IterDataPipe, IterableWrapper
+from torchdata.datapipes.iter import IterableWrapper, IterDataPipe
 
 from ocf_datapipes.batch import MergeNumpyModalities
 from ocf_datapipes.config.model import Configuration
-from ocf_datapipes.load import OpenGSPFromDatabase, OpenPVFromPVSitesDB
+from ocf_datapipes.load import (
+    OpenConfiguration,
+    OpenGSPFromDatabase,
+    OpenPVFromPVSitesDB,
+)
 from ocf_datapipes.training.common import (
     create_t0_and_loc_datapipes,
     open_and_return_datapipes,
@@ -23,9 +27,6 @@
     BatchKey,
     NumpyBatch,
 )
-from ocf_datapipes.load import (
-    OpenConfiguration,
-)
 from ocf_datapipes.utils.utils import combine_to_single_dataset, uncombine_from_single_dataset
 
 xr.set_options(keep_attrs=True)