From f83a1cde04e244b9d5da27f28bcc58daeb4c6b22 Mon Sep 17 00:00:00 2001
From: Chris Briggs <confusedmatrix@gmail.com>
Date: Wed, 28 Feb 2024 14:35:07 +0000
Subject: [PATCH] debug

---
 ocf_datapipes/training/common.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ocf_datapipes/training/common.py b/ocf_datapipes/training/common.py
index 615e64872..5ea5281c6 100644
--- a/ocf_datapipes/training/common.py
+++ b/ocf_datapipes/training/common.py
@@ -470,11 +470,11 @@ def fill_nans_in_arrays(
     Operation is performed in-place on the batch.
     """
     for k, v in batch.items():
+        logger.info(k)
         if isinstance(v, np.ndarray) and np.issubdtype(v.dtype, np.number):
             if np.isnan(v).any():
-                logger.info(k)
                 _filled_keys.update({f"{_key_prefix}{k}"})
-                logger.info('filling nans')
+                logger.info(f'filling nans in {k}')
                 batch[k] = np.nan_to_num(v, copy=True, nan=0.0)
 
         # Recursion is included to reach NWP arrays in subdict