fix w-avg

roocs · Oct 30, 2023 · 267b7f8 · 267b7f8
1 parent a5da5eb
commit 267b7f8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/rook/utils/weighted_average_utils.py b/rook/utils/weighted_average_utils.py
@@ -1,4 +1,6 @@
 import numpy as np
+import xarray as xr
+
 import collections
 
 from roocs_utils.parameter import collection_parameter
@@ -62,9 +64,15 @@ def _calculate(self):
             # add to list
             datasets.append(ds_weighted)
 
+        # concat over time
+        processed_ds = xr.concat(
+            datasets,
+            "time",
+        )
+
         # average
         outputs = average_over_dims(
-            datasets,
+            processed_ds,
             dims=["latitude", "longitude"],
             output_type="nc",
         )