Fix attrs not being kept in spatial_avg()

E3SM-Project · Feb 7, 2024 · fec57c3 · fec57c3
1 parent ea71520
commit fec57c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/e3sm_diags/metrics/metrics.py b/e3sm_diags/metrics/metrics.py
@@ -53,7 +53,7 @@ def spatial_avg(
     weights = _get_weights(ds, var_key, axis)
     dims = _get_dims(dv, axis)
 
-    results = dv.weighted(weights).mean(dims)
+    results = dv.weighted(weights).mean(dims, keep_attrs=True)
 
     if as_list:
         return results.data.tolist()