@@ -122,11 +122,10 @@ use datafusion_comet_proto::{
122122 spark_partitioning:: { partitioning:: PartitioningStruct , Partitioning as SparkPartitioning } ,
123123} ;
124124use datafusion_comet_spark_expr:: {
125- jvm_udf:: JvmScalarUdfExpr ,
126- ArrayInsert , Avg , AvgDecimal , Cast , CheckOverflow , Correlation , Covariance , CreateNamedStruct ,
127- DecimalRescaleCheckOverflow , GetArrayStructFields , GetStructField , IfExpr , ListExtract ,
128- NormalizeNaNAndZero , SparkCastOptions , Stddev , SumDecimal , ToJson , UnboundColumn , Variance ,
129- WideDecimalBinaryExpr , WideDecimalOp ,
125+ jvm_udf:: JvmScalarUdfExpr , ArrayInsert , Avg , AvgDecimal , Cast , CheckOverflow , Correlation ,
126+ Covariance , CreateNamedStruct , DecimalRescaleCheckOverflow , GetArrayStructFields ,
127+ GetStructField , IfExpr , ListExtract , NormalizeNaNAndZero , SparkCastOptions , Stddev , SumDecimal ,
128+ ToJson , UnboundColumn , Variance , WideDecimalBinaryExpr , WideDecimalOp ,
130129} ;
131130use itertools:: Itertools ;
132131use jni:: objects:: { Global , JObject } ;
@@ -708,9 +707,10 @@ impl PhysicalPlanner {
708707 . iter ( )
709708 . map ( |e| self . create_expr ( e, Arc :: clone ( & input_schema) ) )
710709 . collect :: < Result < Vec < _ > , _ > > ( ) ?;
711- let return_type = to_arrow_datatype ( udf. return_type . as_ref ( ) . ok_or_else ( || {
712- GeneralError ( "JvmScalarUdf missing return_type" . to_string ( ) )
713- } ) ?) ;
710+ let return_type =
711+ to_arrow_datatype ( udf. return_type . as_ref ( ) . ok_or_else ( || {
712+ GeneralError ( "JvmScalarUdf missing return_type" . to_string ( ) )
713+ } ) ?) ;
714714 Ok ( Arc :: new ( JvmScalarUdfExpr :: new (
715715 udf. class_name . clone ( ) ,
716716 args,
0 commit comments