From a7df53fd42bbd978bda8ab7f348f2184616c495d Mon Sep 17 00:00:00 2001
From: Jia-Xuan Liu <liugs963@gmail.com>
Date: Wed, 27 Nov 2024 14:53:07 +0800
Subject: [PATCH] fix fmt

---
 wren-core/core/src/mdl/function.rs | 15 +++------------
 wren-core/core/src/mdl/mod.rs      | 13 ++++++++-----
 2 files changed, 11 insertions(+), 17 deletions(-)

diff --git a/wren-core/core/src/mdl/function.rs b/wren-core/core/src/mdl/function.rs
index 13c6b122d..d1505ddde 100644
--- a/wren-core/core/src/mdl/function.rs
+++ b/wren-core/core/src/mdl/function.rs
@@ -71,10 +71,7 @@ impl ByPassScalarUDF {
             name: name.to_string(),
             return_type,
             signature: Signature::one_of(
-                vec![
-                    TypeSignature::VariadicAny,
-                    TypeSignature::NullAry,
-                ],
+                vec![TypeSignature::VariadicAny, TypeSignature::NullAry],
                 Volatility::Volatile,
             ),
         }
@@ -118,10 +115,7 @@ impl ByPassAggregateUDF {
             name: name.to_string(),
             return_type,
             signature: Signature::one_of(
-                vec![
-                    TypeSignature::VariadicAny,
-                    TypeSignature::NullAry,
-                ],
+                vec![TypeSignature::VariadicAny, TypeSignature::NullAry],
                 Volatility::Volatile,
             ),
         }
@@ -165,10 +159,7 @@ impl ByPassWindowFunction {
             name: name.to_string(),
             return_type,
             signature: Signature::one_of(
-                vec![
-                    TypeSignature::VariadicAny,
-                    TypeSignature::NullAry,
-                ],
+                vec![TypeSignature::VariadicAny, TypeSignature::NullAry],
                 Volatility::Volatile,
             ),
         }
diff --git a/wren-core/core/src/mdl/mod.rs b/wren-core/core/src/mdl/mod.rs
index 8f783717f..74685974d 100644
--- a/wren-core/core/src/mdl/mod.rs
+++ b/wren-core/core/src/mdl/mod.rs
@@ -415,7 +415,9 @@ 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::{ArrayRef, Int64Array, RecordBatch, StringArray, TimestampNanosecondArray};
+    use datafusion::arrow::array::{
+        ArrayRef, Int64Array, RecordBatch, StringArray, TimestampNanosecondArray,
+    };
     use datafusion::assert_batches_eq;
     use datafusion::common::not_impl_err;
     use datafusion::common::Result;
@@ -1136,7 +1138,8 @@ mod test {
                         ColumnBuilder::new(
                             "struct_array_col",
                             "array<struct<float_field float,time_field timestamp>>",
-                        ).build(),
+                        )
+                        .build(),
                     )
                     .build(),
             )
@@ -1149,9 +1152,9 @@ mod test {
         (SELECT struct_table.struct_col FROM (SELECT struct_table.struct_col AS struct_col \
         FROM struct_table) AS struct_table) AS struct_table");
 
-
-        let sql  = "select struct_array_col[1].float_field from wren.test.struct_table";
-        let actual = transform_sql_with_ctx(&ctx, Arc::clone(&analyzed_mdl), &[], sql).await?;
+        let sql = "select struct_array_col[1].float_field from wren.test.struct_table";
+        let actual =
+            transform_sql_with_ctx(&ctx, Arc::clone(&analyzed_mdl), &[], sql).await?;
         assert_eq!(actual, "SELECT struct_table.struct_array_col[1].float_field FROM \
         (SELECT struct_table.struct_array_col FROM (SELECT struct_table.struct_array_col AS struct_array_col \
         FROM struct_table) AS struct_table) AS struct_table");