Remove log output of windnet batch

openclimatefix · Feb 29, 2024 · 182ea0e · 182ea0e
1 parent 47e336b
commit 182ea0e
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/ocf_datapipes/training/windnet.py b/ocf_datapipes/training/windnet.py
@@ -166,10 +166,7 @@ def __iter__(self):
             logger.info("Fill in nans")
             combined_datapipe = combined_datapipe.map(fill_nans_in_arrays)
 
-            one_batch = next(iter(combined_datapipe))
-            logger.info(one_batch)
-
-            yield one_batch
+            yield next(iter(combined_datapipe))
 
 
 def minutes(num_mins: int):