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