ahh no it was missing square brakets to define the value

henryaddison · Dec 1, 2023 · d0fc90e · d0fc90e
1 parent 9fb2042
commit d0fc90e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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(