Fix build

systemxlabs · Mar 27, 2024 · 80e0e92 · 80e0e92
1 parent be4df2e
commit 80e0e92
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/function/split.rs b/src/function/split.rs
@@ -1,12 +1,13 @@
 use crate::geo::{GeometryArray, GeometryArrayBuilder};
 use crate::DFResult;
 use arrow_array::cast::AsArray;
-use arrow_array::{BooleanArray, GenericBinaryArray, OffsetSizeTrait};
+use arrow_array::{GenericBinaryArray, OffsetSizeTrait};
 use arrow_schema::DataType;
-use datafusion_common::{internal_datafusion_err, internal_err};
+use datafusion_common::{internal_datafusion_err, internal_err, DataFusionError};
 use datafusion_expr::{ColumnarValue, ScalarUDFImpl, Signature, Volatility};
 use geos::Geom;
 use rayon::iter::IntoParallelIterator;
+use rayon::prelude::*;
 use std::any::Any;
 use std::sync::Arc;
 
@@ -124,7 +125,7 @@ fn split<O: OffsetSizeTrait, F: OffsetSizeTrait>(
             },
         )
         .collect::<DFResult<Vec<Option<geos::Geometry>>>>()?;
-    let builder = GeometryArrayBuilder::<O>::from(&geom_vec);
+    let builder = GeometryArrayBuilder::<O>::from(geom_vec.as_slice());
     Ok(ColumnarValue::Array(Arc::new(builder.build())))
 }
 

diff --git a/src/geo/builder.rs b/src/geo/builder.rs
@@ -192,7 +192,7 @@ impl<O: OffsetSizeTrait> From<&[Option<geo::GeometryCollection>]> for GeometryAr
 }
 
 #[cfg(feature = "geos")]
-impl<O: OffsetSizeTrait> From<&[Option<geos::Geometry>]> for GeometryArrayBuilder<O> {
+impl<O: OffsetSizeTrait> From<&[Option<geos::Geometry<'_>>]> for GeometryArrayBuilder<O> {
     fn from(value: &[Option<geos::Geometry>]) -> Self {
         let mut builder = GeometryArrayBuilder::<O>::new(WkbDialect::Ewkb, value.len());
         for geom in value {