Merge pull request #277 from openclimatefix/issue/more-logging

add logging

Author: peterdudfield

ocf_datapipes/training/windnet.py

@@ -163,9 +163,13 @@ def __iter__(self):
             # in production we don't want any nans in the satellite data
             #    combined_datapipe = combined_datapipe.map(check_nans_in_satellite_data)
 
+            logger.info("Fill in nans")
             combined_datapipe = combined_datapipe.map(fill_nans_in_arrays)
 
-            yield next(iter(combined_datapipe))
+            one_batch = next(iter(combined_datapipe))
+            logger.info(one_batch)
+
+            yield one_batch
 
 
 def minutes(num_mins: int):