From 451381c5707d976c326fb9d2f5910d16cd33821a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 5 Dec 2023 10:15:33 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- pvnet/data/wind_datamodule.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/pvnet/data/wind_datamodule.py b/pvnet/data/wind_datamodule.py index f4008d10..aac36a58 100644 --- a/pvnet/data/wind_datamodule.py +++ b/pvnet/data/wind_datamodule.py @@ -1,6 +1,5 @@ """ Data module for pytorch lightning """ import glob -from datetime import datetime from lightning.pytorch import LightningDataModule from ocf_datapipes.training.windnet import windnet_netcdf_datapipe @@ -46,19 +45,19 @@ def __init__( self.batch_size = batch_size self.batch_dir = batch_dir - #if batch_dir is not None: + # if batch_dir is not None: # if any([period != [None, None] for period in [train_period, val_period, test_period]]): # raise ValueError("Cannot set `(train/val/test)_period` with presaved batches") self.train_period = [None, None] # None if d is None else datetime.strptime(d, "%Y-%m-%d") for d in train_period - #] + # ] self.val_period = [None, None] # None if d is None else datetime.strptime(d, "%Y-%m-%d") for d in val_period - #] + # ] self.test_period = [None, None] # None if d is None else datetime.strptime(d, "%Y-%m-%d") for d in test_period - #] + # ] self._common_dataloader_kwargs = dict( shuffle=False, # shuffled in datapipe step