OK got to expand dims for entire dataset

henryaddison · Dec 1, 2023 · 9fb2042 · 9fb2042
1 parent a2f033c
commit 9fb2042
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ml_downscaling_emulator/bin/postprocess.py b/src/ml_downscaling_emulator/bin/postprocess.py
@@ -180,7 +180,7 @@ def gcmify(
 ):
     def process_samples(ds):
         ds = to_gcm_domain(ds.sel(ensemble_member=ensemble_member))
-        ds["pred_pr"] = ds["pred_pr"].expand_dims({"ensemble_member": ensemble_member})
+        ds = ds.expand_dims({"ensemble_member": ensemble_member})
         return ds
 
     process_each_sample(