From 7f671ee71a674139fca161d29ea6f26b577be99f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Oriol=20Tint=C3=B3?= <oriol.tinto@lmu.de>
Date: Tue, 20 Jun 2023 14:45:45 +0200
Subject: [PATCH] Fix integrate_sst_to_analysis
---
templates/event-generator/adapt_member.sh | 2 +-
templates/event-generator/adapt_member_utils.sh | 9 ++++-----
2 files changed, 5 insertions(+), 6 deletions(-)
diff --git a/templates/event-generator/adapt_member.sh b/templates/event-generator/adapt_member.sh
index 8f0bc64..3a552ad 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 4c1759a..a1635b5 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
--
GitLab