debug

Author: confusedmatrix

ocf_datapipes/training/common.py

@@ -472,9 +472,10 @@ def fill_nans_in_arrays(
     for k, v in batch.items():
         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}"})
-                # batch[k] = np.nan_to_num(v, copy=True, nan=0.0)
-                np.nan_to_num(v, copy=False, nan=0.0)
+                logger.info('filling nans')
+                batch[k] = np.nan_to_num(v, copy=True, nan=0.0)
 
         # Recursion is included to reach NWP arrays in subdict
         elif isinstance(v, dict):