From f24cf92ecd9f18dbd690747efb6188cee5b3d50b Mon Sep 17 00:00:00 2001 From: peterdudfield Date: Tue, 16 Nov 2021 18:01:10 +0000 Subject: [PATCH] change nwp float32 --- nowcasting_dataset/data_sources/nwp/nwp_data_source.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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