Fix GFI when vertical coords are specified (#36)

* Fix GFI when vertical coords are specified

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update target versions since dependencies havent all bee updated

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Udpate python version for packaging

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

.github/workflows/publish-to-pypi.yml

@@ -22,7 +22,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v4
         with:
-          python-version: "3.x"
+          python-version: "3.11.x"
 
       # - name: Get tags
       #   run: git fetch --depth=1 origin +refs/tags/*:refs/tags/*

.gitignore

@@ -9,3 +9,4 @@ __pycache__/
 build/
 dist/
 xpublish_wms/_version.py
+env/

pyproject.toml

@@ -7,7 +7,7 @@ name = "xpublish_wms"
 authors = [{ name = "Matthew Iannucci", email = "[email protected]" }]
 description = "WMS plugin for xpublish"
 readme = "README.md"
-requires-python = ">=3.9"
+requires-python = ">=3.9,<3.12"
 keywords = ["xarray", "xpublish", "wms"]
 license = { file = "LICENSE.txt" }
 
@@ -21,6 +21,7 @@ classifiers = [
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
     "Topic :: Scientific/Engineering",
 ]
 

xpublish_wms/grid.py

@@ -100,7 +100,13 @@ def tessellate(self, da: xr.DataArray) -> np.ndarray:
         """Tessellate the given data array into triangles. Only required for RenderingMode.Triangle"""
         pass
 
-    def sel_lat_lng(self, subset: xr.Dataset, lng, lat, parameters) -> Tuple[xr.Dataset, list, list]:
+    def sel_lat_lng(
+        self,
+        subset: xr.Dataset,
+        lng,
+        lat,
+        parameters,
+    ) -> Tuple[xr.Dataset, list, list]:
         """Select the given dataset by the given lon/lat and optional elevation"""
         subset = subset.cf.interp(longitude=lng, latitude=lat)
         x_axis = [strip_float(subset.cf["longitude"])]
@@ -183,7 +189,13 @@ def project(self, da: xr.DataArray, crs: str) -> xr.DataArray:
             da = da.unify_chunks()
         return da
 
-    def sel_lat_lng(self, subset: xr.Dataset, lng, lat, parameters) -> Tuple[xr.Dataset, list, list]:
+    def sel_lat_lng(
+        self,
+        subset: xr.Dataset,
+        lng,
+        lat,
+        parameters,
+    ) -> Tuple[xr.Dataset, list, list]:
         topology = self.ds.cf["grid_topology"]
 
         merged_ds = None
@@ -515,7 +527,13 @@ def tessellate(self, da: xr.DataArray) -> np.ndarray:
         else:
             return self._grid.tessellate(da)
 
-    def sel_lat_lng(self, subset: xr.Dataset, lng, lat, parameters) -> Tuple[xr.Dataset, list, list]:
+    def sel_lat_lng(
+        self,
+        subset: xr.Dataset,
+        lng,
+        lat,
+        parameters,
+    ) -> Tuple[xr.Dataset, list, list]:
         if self._grid is None:
             return None
         else:

xpublish_wms/wms/get_feature_info.py

@@ -161,15 +161,20 @@ def get_feature_info(ds: xr.Dataset, query: dict) -> Response:
             # Dont select an elevation, just keep all elevation coords
             elevation = selected_ds.cf["vertical"].values
         elif len(elevation) == 1:
-            selected_ds = selected_ds.cf.interp(vertical=elevation)
+            selected_ds = selected_ds.cf.sel(vertical=elevation, method="nearest")
         elif len(elevation) > 1:
             selected_ds = selected_ds.cf.sel(vertical=slice(elevation[0], elevation[1]))
         else:
             # Select closest to the surface by default
             selected_ds = selected_ds.cf.sel(vertical=0, method="nearest")
 
     try:
-        selected_ds, x_axis, y_axis = ds.grid.sel_lat_lng(subset=selected_ds, lng=x_coord[x], lat=y_coord[y], parameters=parameters)
+        selected_ds, x_axis, y_axis = ds.grid.sel_lat_lng(
+            subset=selected_ds,
+            lng=x_coord[x],
+            lat=y_coord[y],
+            parameters=parameters,
+        )
     except ValueError:
         raise HTTPException(500, f"Unsupported grid type: {ds.grid.name}")