Nans updated in place

Author: confusedmatrix

ocf_datapipes/training/common.py

@@ -474,6 +474,7 @@ def fill_nans_in_arrays(
             if np.isnan(v).any():
                 _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)
 
         # Recursion is included to reach NWP arrays in subdict
         elif isinstance(v, dict):