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

Author: hongyuchen1030

uxarray/core/dataarray.py

@@ -483,6 +483,13 @@ 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/grid/grid.py

@@ -1455,6 +1455,15 @@ def faces_edges_cartesian(self):
             _populate_faces_edges_cartesian(self)
         return self._ds["faces_edges_cartesian"]
 
+    @faces_edges_cartesian.setter
+    def faces_edges_cartesian(self, value):
+        self._ds["faces_edges_cartesian"] = value
+
+    @faces_edges_cartesian.deleter
+    def faces_edges_cartesian(self):
+        if "faces_edges_cartesian" in self._ds:
+            del self._ds["faces_edges_cartesian"]
+
     @property
     def faces_edges_spherical(self):
         """Latitude Longitude Coordinates for each Face in radians.
@@ -1465,6 +1474,15 @@ def faces_edges_spherical(self):
             _populate_faces_edges_spherical(self)
         return self._ds["faces_edges_spherical"]
 
+    @faces_edges_spherical.setter
+    def faces_edges_spherical(self, value):
+        self._ds["faces_edges_spherical"] = value
+
+    @faces_edges_spherical.deleter
+    def faces_edges_spherical(self):
+        if "faces_edges_spherical" in self._ds:
+            del self._ds["faces_edges_spherical"]
+
     @property
     def bounds(self):
         """Latitude Longitude Bounds for each Face in radians.