keep attributes

roocs · Nov 8, 2023 · a2ba23e · a2ba23e
1 parent 4802b55
commit a2ba23e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rook/utils/weighted_average_utils.py b/rook/utils/weighted_average_utils.py
@@ -27,7 +27,7 @@ def calc_weighted_mean(ds):
     # apply weights
     ds_weighted = ds.weighted(weights)
     # apply mean
-    ds_weighted_mean = ds_weighted.mean(["lat", "lon"])
+    ds_weighted_mean = ds_weighted.mean(["lat", "lon"], keep_attrs=True)
     return ds_weighted_mean