make sure sat and nwp are saved as np.float64

Author: peterdudfield

nowcasting_dataset/data_sources/nwp/nwp_data_source.py

@@ -233,6 +233,9 @@ def _post_process_example(
         selected_data = selected_data.sel(target_time=slice(start_dt, end_dt))
         selected_data = selected_data.rename({"target_time": "time"})
         selected_data = selected_data.rename({"variable": "channels"})
+
+        selected_data.data = selected_data.data.astype(np.float32)
+
         return selected_data
 
     def datetime_index(self) -> pd.DatetimeIndex:

nowcasting_dataset/data_sources/satellite/satellite_data_source.py

@@ -175,6 +175,9 @@ def _post_process_example(
         if self.normalise:
             selected_data = selected_data - SAT_MEAN
             selected_data = selected_data / SAT_STD
+
+        selected_data.data = selected_data.data.astype(np.float32)
+
         return selected_data
 
     def datetime_index(self, remove_night: bool = True) -> pd.DatetimeIndex: