From d6df7f7d8efbe02a5b581636c8b7809fa58aee22 Mon Sep 17 00:00:00 2001 From: James Date: Wed, 20 Dec 2023 18:29:08 +0000 Subject: [PATCH] update imports --- pvnet_summation/data/datamodule.py | 2 +- tests/conftest.py | 2 +- tests/data/test_datamodule.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pvnet_summation/data/datamodule.py b/pvnet_summation/data/datamodule.py index 799a3e2..175e705 100644 --- 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 index f22a0f2..98a13fe 100644 --- 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 index 540133d..5c6ca72 100644 --- 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):