Fix python bindings

geoarrow · Dec 23, 2024 · e5a9f57 · e5a9f57
1 parent 5e4f644
commit e5a9f57
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 51 deletions.
diff --git a/python/Cargo.lock b/python/Cargo.lock
diff --git a/python/pyo3-geoarrow/src/ffi/from_python/scalar.rs b/python/pyo3-geoarrow/src/ffi/from_python/scalar.rs
@@ -2,8 +2,7 @@ use std::sync::Arc;
 
 use crate::array::*;
 use crate::scalar::*;
-use geoarrow::datatypes::Dimension;
-use geoarrow::io::geozero::ToMixedArray;
+use geoarrow::io::geozero::ToGeometryArray;
 use geoarrow::scalar::GeometryScalar;
 use geozero::geojson::GeoJsonString;
 use pyo3::exceptions::PyValueError;
@@ -24,9 +23,9 @@ impl<'a> FromPyObject<'a> for PyGeometry {
             // Parse GeoJSON to geometry scalar
             let reader = GeoJsonString(json_string);
 
-            // TODO: we need a dynamic dimensionality reader
+            // TODO: use ToGeometry directly in the future?
             let arr = reader
-                .to_mixed_geometry_array(Dimension::XY)
+                .to_geometry_array()
                 .map_err(|err| PyValueError::new_err(err.to_string()))?;
             Ok(Self(
                 GeometryScalar::try_new(Arc::new(arr))