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

add logging
openclimatefix · Feb 27, 2024 · 4e28e76 · 4e28e76
2 parents 78e0f0b + b593b72
commit 4e28e76
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ocf_datapipes/training/windnet.py b/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):