feat: update geopandas notebook

Author: gadomski

Cargo.lock

@@ -12,7 +12,7 @@ crate-type = ["cdylib"]
 clap = "4.5.31"
 geojson = "0.24.1"
 geoarrow = "0.4.0-beta.3"
-pyo3 = { version = "0.23.4", features = ["extension-module"] }
+pyo3 = { version = "0.23.5", features = ["extension-module"] }
 pyo3-async-runtimes = { version = "0.23.0", features = [
     "tokio",
     "tokio-runtime",

Cargo.toml

@@ -36,9 +36,19 @@ pub fn to_arrow(py: Python<'_>, items: Bound<PyAny>) -> PyResult<PyObject> {
     } else {
         serde_json::from_value(value).map_err(Error::from)?
     };
-    let (record_batches, schema) = stac::geoarrow::to_table(item_collection)
+    // TODO we might want to just allow use to go WKB right when we got to table?
+    let (record_batches, mut schema) = stac::geoarrow::to_table(item_collection)
         .map_err(Error::from)?
         .into_inner();
+    let record_batches = record_batches
+        .into_iter()
+        .map(|record_batch| {
+            stac::geoarrow::with_wkb_geometry(record_batch, "geometry").map_err(Error::from)
+        })
+        .collect::<Result<Vec<_>>>()?;
+    if !record_batches.is_empty() {
+        schema = record_batches[0].schema();
+    }
     let table = PyTable::try_new(record_batches, schema)?;
     let table = table.to_arro3(py)?;
     Ok(table.into_py_any(py)?)