Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 5, 2023
1 parent 3370353 commit 451381c
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions pvnet/data/wind_datamodule.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 451381c

Please sign in to comment.