From d0fc90eb1de90b54bb6ca8d087cdb771acfa24ed Mon Sep 17 00:00:00 2001 From: Henry Addison Date: Fri, 1 Dec 2023 16:12:58 +0000 Subject: [PATCH] ahh no it was missing square brakets to define the value --- src/ml_downscaling_emulator/bin/postprocess.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/ml_downscaling_emulator/bin/postprocess.py b/src/ml_downscaling_emulator/bin/postprocess.py index 75a0da97c..2eb4e5c2d 100644 --- a/src/ml_downscaling_emulator/bin/postprocess.py +++ b/src/ml_downscaling_emulator/bin/postprocess.py @@ -180,7 +180,9 @@ def gcmify( ): def process_samples(ds): ds = to_gcm_domain(ds.sel(ensemble_member=ensemble_member)) - ds = ds.expand_dims({"ensemble_member": ensemble_member}) + ds["pred_pr"] = ds["pred_pr"].expand_dims( + {"ensemble_member": [ensemble_member]} + ) return ds process_each_sample(