add logging

Author: peterdudfield

ocf_datapipes/transform/xarray/add_t0idx_and_sample_period_duration.py

@@ -1,11 +1,14 @@
 """Adds useful t0_idx and sample period attributes"""
 from datetime import timedelta
 from typing import Union
+import logging
 
 import xarray as xr
 from torchdata.datapipes import functional_datapipe
 from torchdata.datapipes.iter import IterDataPipe
 
+logger = logging.getLogger(__name__)
+
 
 @functional_datapipe("add_t0_idx_and_sample_period_duration")
 class AddT0IdxAndSamplePeriodDurationIterDataPipe(IterDataPipe):
@@ -33,6 +36,7 @@ def __init__(
     def __iter__(self) -> Union[xr.DataArray, xr.Dataset]:
         """Adds the two attributes to the xarray objects and returns them"""
         for xr_data in self.source_datapipe:
+            logger.debug("Adding t0 and sample_period_duration to xarray")
             xr_data.attrs["t0_idx"] = self.t0_idx
             xr_data.attrs["sample_period_duration"] = self.sample_period_duration
             yield xr_data

tests/conftest.py

@@ -231,7 +231,6 @@ def gsp_yields(db_session):
     for gsp_id in range(1,3):
         gsp_sql_1: LocationSQL = Location(gsp_id=gsp_id, label="GSP_1", installed_capacity_mw=1).to_orm()
 
-        gsp_yield_sqls = []
         for hour in range(0, 8):
             for minute in range(0, 60, 30):
                 gsp_yield_1 = GSPYield(