style: apply rustfmt formatting

Co-authored-by: Isaac

Author: Yu Chen

crates/lance-graph/src/query.rs

@@ -385,13 +385,12 @@ impl CypherQuery {
         // Unparse to SQL using the specified dialect
         let dialect_unparser = dialect.unparser();
         let unparser = dialect_unparser.as_unparser();
-        let sql_ast =
-            unparser
-                .plan_to_sql(&optimized_plan)
-                .map_err(|e| GraphError::PlanError {
-                    message: format!("Failed to unparse plan to SQL: {}", e),
-                    location: snafu::Location::new(file!(), line!(), column!()),
-                })?;
+        let sql_ast = unparser
+            .plan_to_sql(&optimized_plan)
+            .map_err(|e| GraphError::PlanError {
+                message: format!("Failed to unparse plan to SQL: {}", e),
+                location: snafu::Location::new(file!(), line!(), column!()),
+            })?;
 
         Ok(sql_ast.to_string())
     }

crates/lance-graph/src/spark_dialect.rs

@@ -15,10 +15,10 @@
 //! - `LENGTH()` instead of `CHARACTER_LENGTH()`
 //! - Subqueries in FROM require aliases
 
+use datafusion_sql::sqlparser::ast::{self, Ident, ObjectName, TimezoneInfo};
 use datafusion_sql::unparser::dialect::{
     CharacterLengthStyle, CustomDialect, CustomDialectBuilder, DateFieldExtractStyle,
 };
-use datafusion_sql::sqlparser::ast::{self, Ident, ObjectName, TimezoneInfo};
 
 /// Build a Spark SQL dialect using DataFusion's `CustomDialectBuilder`.
 pub fn build_spark_dialect() -> CustomDialect {
@@ -66,9 +66,18 @@ mod tests {
     #[test]
     fn test_spark_dialect_type_mappings() {
         let dialect = build_spark_dialect();
-        assert!(matches!(dialect.utf8_cast_dtype(), ast::DataType::Custom(..)));
-        assert!(matches!(dialect.int64_cast_dtype(), ast::DataType::BigInt(None)));
-        assert!(matches!(dialect.int32_cast_dtype(), ast::DataType::Int(None)));
+        assert!(matches!(
+            dialect.utf8_cast_dtype(),
+            ast::DataType::Custom(..)
+        ));
+        assert!(matches!(
+            dialect.int64_cast_dtype(),
+            ast::DataType::BigInt(None)
+        ));
+        assert!(matches!(
+            dialect.int32_cast_dtype(),
+            ast::DataType::Int(None)
+        ));
         assert!(matches!(dialect.date32_cast_dtype(), ast::DataType::Date));
     }