debug

Author: confusedmatrix

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