diff --git a/nowcasting_dataset/data_sources/nwp/nwp_data_source.py b/nowcasting_dataset/data_sources/nwp/nwp_data_source.py index cd18a957..02d0bf66 100644 --- a/nowcasting_dataset/data_sources/nwp/nwp_data_source.py +++ b/nowcasting_dataset/data_sources/nwp/nwp_data_source.py @@ -130,7 +130,7 @@ def _post_process_example(self, selected_data: xr.Dataset, t0_dt: pd.Timestamp) selected_data = selected_data.sel(target_time=slice(start_dt, end_dt)) selected_data = selected_data.rename({"target_time": "time", "variable": "channels"}) - selected_data.data = selected_data.data.astype(np.float16) + selected_data.data = selected_data.data.astype(np.float32) return selected_data