diff --git a/templates/event-generator/adapt_member.sh b/templates/event-generator/adapt_member.sh index 8f0bc641718b8931f11002dab5751066ae9c85ff..3a552adbdf3763a15613fc861867de62614e87d8 100644 --- a/templates/event-generator/adapt_member.sh +++ b/templates/event-generator/adapt_member.sh @@ -10,7 +10,7 @@ WORKDIR=%HPCROOTDIR% PROJ_FOLDER=${WORKDIR}/production_project SST_FORCING=$( basename %simulation.initial_conditions.sea_surface_temperature_forcing% ) -INTERPOLATED_SST=whatever.nc +INTERPOLATED_SST=sst_climatology.nc DESTINATION_GRID=%simulation.dynamics_grid_filename% EXTERNAL_PARAMETERS_FILE=%simulation.external_parameters_filename% STARTDATE=%SDATE% diff --git a/templates/event-generator/adapt_member_utils.sh b/templates/event-generator/adapt_member_utils.sh index 4c1759a06a807bc356ea7cbfb0178d047a11a3cc..a1635b5e01338c303d5fd51238005cbb2f8056fb 100644 --- a/templates/event-generator/adapt_member_utils.sh +++ b/templates/event-generator/adapt_member_utils.sh @@ -99,18 +99,17 @@ function integrate_sst_to_analysis() { cat <<END > integrate_sst_to_analysis.py # Add sst field to analysis file. -import xarray as xr +from enstools.io import read import os import numpy as np os.environ['ECCODES_GRIB_NO_INDEX'] = '1' # Open files -with xr.open_dataset("${ANALYSIS_FILE}") as ds, xr.open_dataset("${INTERPOLATED_SST}") as sst_ds: +with read("${ANALYSIS_FILE}") as ds, read("${INTERPOLATED_SST}") as sst_ds: date=ds.time.values - day_of_year = int((date - np.datetime64(date.astype('datetime64[Y]'))) / np.timedelta64(1, 'D')) - + day_of_year = int((date - date.astype('datetime64[Y]')) / np.timedelta64(1, 'D')) # Replace analysis T_SEA with info from the sst_clim file. - ds["T_SEA"]= sst_ds.isel(time=day_of_year+1) + ds["T_SEA"]= sst_ds["T_SEA"].isel(time=day_of_year+1) ds.to_netcdf("analysis.nc") END