use arrow functions to extract cell coordinates (#114)

* expect `cells_to_coordinates` to return a `RecordBatch` * also use `assert_allclose` for the h3 tutorial * changelog
xarray-contrib · Dec 9, 2024 · 82c3adf · 82c3adf
1 parent 0e7c5e0
commit 82c3adf
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/docs/changelog.md b/docs/changelog.md
@@ -6,6 +6,8 @@
 
 ## Bug fixes
 
+- use explicit `arrow` API to extract cell coordinates ({issue}`113`, {pull}`114`)
+
 ## Documentation
 
 ## Internal changes

diff --git a/docs/tutorials/h3.ipynb b/docs/tutorials/h3.ipynb
@@ -150,7 +150,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "xr.testing.assert_equal(derived_ds, original_ds)"
+    "xr.testing.assert_allclose(derived_ds, original_ds)"
    ]
   },
   {

diff --git a/xdggs/h3.py b/xdggs/h3.py
@@ -141,7 +141,10 @@ def cell_ids2geographic(
         lat : array-like
             The latitude coordinate values of the grid cells in degree
         """
-        lat, lon = cells_to_coordinates(cell_ids, radians=False)
+        result = cells_to_coordinates(cell_ids, radians=False)
+
+        lon = result.column("lng").to_numpy()
+        lat = result.column("lat").to_numpy()
 
         return lon, lat