@@ -586,7 +586,7 @@ where
586586 . ok_or_else ( || {
587587 sedona_internal_datafusion_err ! ( "Expected StructArray for raster data" )
588588 } ) ?;
589- let raster_array = RasterStructArray :: new ( raster_struct) ;
589+ let raster_array = RasterStructArray :: try_new ( raster_struct) ? ;
590590
591591 for i in 0 ..num_iterations {
592592 if raster_array. is_null ( i) {
@@ -600,7 +600,7 @@ where
600600 Ok ( ( ) )
601601 }
602602 ColumnarValue :: Scalar ( ScalarValue :: Struct ( raster_struct) ) => {
603- let raster_array = RasterStructArray :: new ( raster_struct. as_ref ( ) ) ;
603+ let raster_array = RasterStructArray :: try_new ( raster_struct. as_ref ( ) ) ? ;
604604 let raster = if raster_array. is_null ( 0 ) {
605605 None
606606 } else {
@@ -660,7 +660,7 @@ mod tests {
660660
661661 fn load_reference_raster ( ) -> RasterMetadata {
662662 let raster_array = reference_raster_spec ( ) . build ( ) ;
663- let raster_struct = RasterStructArray :: new ( & raster_array) ;
663+ let raster_struct = RasterStructArray :: try_new ( & raster_array) . unwrap ( ) ;
664664 raster_struct. get ( 0 ) . unwrap ( ) . metadata ( )
665665 }
666666
@@ -681,7 +681,7 @@ mod tests {
681681 fn test_rs_as_raster_use_reference_extent ( ) {
682682 let raster_array = reference_raster_spec ( ) . build ( ) ;
683683 with_gdal ( |gdal| {
684- let raster_struct = RasterStructArray :: new ( & raster_array) ;
684+ let raster_struct = RasterStructArray :: try_new ( & raster_array) . unwrap ( ) ;
685685 let raster = raster_struct. get ( 0 ) . unwrap ( ) ;
686686 let md = raster. metadata ( ) ;
687687
@@ -719,7 +719,7 @@ mod tests {
719719 fn test_rs_as_raster_use_geometry_extent ( ) {
720720 let raster_array = reference_raster_spec ( ) . build ( ) ;
721721 with_gdal ( |gdal| {
722- let raster_struct = RasterStructArray :: new ( & raster_array) ;
722+ let raster_struct = RasterStructArray :: try_new ( & raster_array) . unwrap ( ) ;
723723 let raster = raster_struct. get ( 0 ) . unwrap ( ) ;
724724 let md = raster. metadata ( ) ;
725725
0 commit comments