add explainPlan info and fix fmt

huaxingao · Apr 24, 2024 · 862d20a · 862d20a
1 parent 9c1c808
commit 862d20a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/core/src/execution/datafusion/planner.rs b/core/src/execution/datafusion/planner.rs
@@ -1248,11 +1248,11 @@ impl PhysicalPlanner {
                         expr.null_on_divide_by_zero,
                     ))),
                     1 => Ok(Arc::new(Variance::new(
-                         child,
-                         "variance_pop",
-                         datatype,
-                         StatsType::Population,
-                         expr.null_on_divide_by_zero,
+                        child,
+                        "variance_pop",
+                        datatype,
+                        StatsType::Population,
+                        expr.null_on_divide_by_zero,
                     ))),
                     stats_type => Err(ExecutionError::GeneralError(format!(
                         "Unknown StatisticsType {:?} for Variance",

diff --git a/spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala b/spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala
@@ -481,6 +481,7 @@ object QueryPlanSerde extends Logging with ShimQueryPlanSerde {
               .setVariance(varBuilder)
               .build())
         } else {
+          withInfo(aggExpr, child)
           None
         }
       case variancePop @ VariancePop(child, nullOnDivideByZero) =>
@@ -500,6 +501,7 @@ object QueryPlanSerde extends Logging with ShimQueryPlanSerde {
               .setVariance(varBuilder)
               .build())
         } else {
+          withInfo(aggExpr, child)
           None
         }
       case fn =>