upgrade type signature

Canner · Nov 27, 2024 · 5769c31 · 5769c31
1 parent ef3bcda
commit 5769c31
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/wren-core/core/src/mdl/function.rs b/wren-core/core/src/mdl/function.rs
@@ -73,7 +73,7 @@ impl ByPassScalarUDF {
             signature: Signature::one_of(
                 vec![
                     TypeSignature::VariadicAny,
-                    TypeSignature::Uniform(0, vec![]),
+                    TypeSignature::NullAry,
                 ],
                 Volatility::Volatile,
             ),
@@ -120,7 +120,7 @@ impl ByPassAggregateUDF {
             signature: Signature::one_of(
                 vec![
                     TypeSignature::VariadicAny,
-                    TypeSignature::Uniform(0, vec![]),
+                    TypeSignature::NullAry,
                 ],
                 Volatility::Volatile,
             ),
@@ -167,7 +167,7 @@ impl ByPassWindowFunction {
             signature: Signature::one_of(
                 vec![
                     TypeSignature::VariadicAny,
-                    TypeSignature::Uniform(0, vec![]),
+                    TypeSignature::NullAry,
                 ],
                 Volatility::Volatile,
             ),

diff --git a/wren-core/core/src/mdl/mod.rs b/wren-core/core/src/mdl/mod.rs
@@ -415,9 +415,7 @@ mod test {
     use crate::mdl::function::RemoteFunction;
     use crate::mdl::manifest::Manifest;
     use crate::mdl::{self, transform_sql_with_ctx, AnalyzedWrenMDL};
-    use datafusion::arrow::array::{Array, ArrayData, ArrayRef, Float64Array, Int64Array, ListArray, RecordBatch, StringArray, StructArray, TimestampNanosecondArray};
-    use datafusion::arrow::buffer::Buffer;
-    use datafusion::arrow::datatypes::{DataType, Field, Fields, TimeUnit};
+    use datafusion::arrow::array::{ArrayRef, Int64Array, RecordBatch, StringArray, TimestampNanosecondArray};
     use datafusion::assert_batches_eq;
     use datafusion::common::not_impl_err;
     use datafusion::common::Result;