Remove the "faces_edges_cartesian" and "faces_edges_spherical" at the end of zonal_avg

Author: hongyuchen1030

uxarray/core/dataarray.py

@@ -483,13 +483,6 @@ def zonal_mean(self, lat=(-90, 90, 10), **kwargs):
             uxda=self, latitudes=latitudes, **kwargs
         )
 
-        # Remove the "faces_edges_cartesian" and "faces_edges_spherical" since they're no longer needed
-        if "faces_edges_cartesian" in self._ds:
-            del self.faces_edges_cartesian
-
-        if "faces_edges_spherical" in self._ds:
-            del self.faces_edges_spherical
-
         dims = list(self.dims[:-1]) + ["latitudes"]
 
         uxda = UxDataArray(

uxarray/core/zonal.py

@@ -48,4 +48,10 @@ def _compute_non_conservative_zonal_mean(uxda, latitudes, use_robust_weights=Fal
             axis=-1
         )
 
+    # Remove the "faces_edges_cartesian" and "faces_edges_spherical" since they're no longer needed
+    if "faces_edges_cartesian" in uxgrid._ds:
+        del uxgrid.faces_edges_cartesian
+
+    if "faces_edges_spherical" in uxgrid._ds:
+        del uxgrid.faces_edges_spherical
     return result