fix origin and extent on lon/lat variables in MetricTerms (#330)

## Purpose This PR fixes the origin and extent on lat, lon, lat_agrid, and lon_agrid attributes of MetricTerms.
ai2cm · Oct 7, 2022 · f18e940 · f18e940
1 parent 265176c
commit f18e940
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/util/pace/util/grid/generation.py b/util/pace/util/grid/generation.py
@@ -258,7 +258,9 @@ def agrid_lon_lat(self):
     def lon(self):
         return util.Quantity(
             data=self.grid.data[:, :, 0],
-            dims=self.grid.dims[0:2],
+            dims=self.grid.dims[:2],
+            origin=self.grid.origin[:2],
+            extent=self.grid.extent[:2],
             units=self.grid.units,
             gt4py_backend=self.grid.gt4py_backend,
         )
@@ -267,7 +269,9 @@ def lon(self):
     def lat(self) -> util.Quantity:
         return util.Quantity(
             data=self.grid.data[:, :, 1],
-            dims=self.grid.dims[0:2],
+            dims=self.grid.dims[:2],
+            origin=self.grid.origin[:2],
+            extent=self.grid.extent[:2],
             units=self.grid.units,
             gt4py_backend=self.grid.gt4py_backend,
         )
@@ -276,7 +280,9 @@ def lat(self) -> util.Quantity:
     def lon_agrid(self) -> util.Quantity:
         return util.Quantity(
             data=self.agrid.data[:, :, 0],
-            dims=self.agrid.dims[0:2],
+            dims=self.agrid.dims[:2],
+            origin=self.agrid.origin[:2],
+            extent=self.agrid.extent[:2],
             units=self.agrid.units,
             gt4py_backend=self.agrid.gt4py_backend,
         )
@@ -285,7 +291,9 @@ def lon_agrid(self) -> util.Quantity:
     def lat_agrid(self) -> util.Quantity:
         return util.Quantity(
             data=self.agrid.data[:, :, 1],
-            dims=self.agrid.dims[0:2],
+            dims=self.agrid.dims[:2],
+            origin=self.agrid.origin[:2],
+            extent=self.agrid.extent[:2],
             units=self.agrid.units,
             gt4py_backend=self.agrid.gt4py_backend,
         )