Merge commit 'd6df7f7d8efbe02a5b581636c8b7809fa58aee22' into issue/rm…

…-nowcasting-utils
openclimatefix · Dec 20, 2023 · 0bade7d · 0bade7d
2 parents f8ac5a2 + d6df7f7
commit 0bade7d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pvnet_summation/data/datamodule.py b/pvnet_summation/data/datamodule.py
@@ -4,7 +4,7 @@
 from lightning.pytorch import LightningDataModule
 from ocf_datapipes.load import OpenGSP
 from ocf_datapipes.training.pvnet import normalize_gsp
-from ocf_datapipes.utils.consts import BatchKey
+from ocf_datapipes.batch import BatchKey
 from torch.utils.data import DataLoader
 from torch.utils.data.datapipes.datapipe import IterDataPipe
 from torch.utils.data.datapipes.iter import FileLister, Zipper

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -11,7 +11,7 @@
 from pvnet_summation.models.model import Model
 
 
-from ocf_datapipes.utils.consts import BatchKey
+from ocf_datapipes.batch import BatchKey
 from datetime import timedelta
 
 from pvnet_summation.data.datamodule import DataModule

diff --git a/tests/data/test_datamodule.py b/tests/data/test_datamodule.py
@@ -1,5 +1,5 @@
 from pvnet_summation.data.datamodule import DataModule
-from ocf_datapipes.utils.consts import BatchKey
+from ocf_datapipes.batch import BatchKey
 
 
 def test_init(sample_data):