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

openclimatefix · Feb 5, 2024 · a240276 · a240276
2 parents 98a0acb + bc9def1
commit a240276
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/save_batches.py b/scripts/save_batches.py
@@ -26,6 +26,7 @@
 import dask
 import hydra
 import torch
+from ocf_datapipes.batch import stack_np_examples_into_batch
 from ocf_datapipes.training.pvnet import pvnet_datapipe
 from ocf_datapipes.training.pvnet_site import pvnet_site_datapipe
 from ocf_datapipes.training.windnet import windnet_datapipe
@@ -34,9 +35,8 @@
 from torch.utils.data import DataLoader
 from torch.utils.data.datapipes.iter import IterableWrapper
 from tqdm import tqdm
-from ocf_datapipes.batch import stack_np_examples_into_batch
-from pvnet.data.utils import batch_to_tensor
 
+from pvnet.data.utils import batch_to_tensor
 from pvnet.utils import print_config
 
 dask.config.set(scheduler="single-threaded")
@@ -77,7 +77,7 @@ def _get_datapipe(config_path, start_time, end_time, batch_size, renewable: str
     )
     if renewable == "pv":
         data_pipeline = (
-           data_pipeline.batch(batch_size).map(stack_np_examples_into_batch).map(batch_to_tensor)
+            data_pipeline.batch(batch_size).map(stack_np_examples_into_batch).map(batch_to_tensor)
         )
     return data_pipeline