fix

Author: viirya

native/spark-expr/src/variance.rs

@@ -245,10 +245,8 @@ impl Accumulator for VarianceAccumulator {
 
         Ok(ScalarValue::Float64(match self.count {
             count if count == 0.0 => None,
-            count if count == 1.0 => {
-                if let StatsType::Population = self.stats_type {
-                    Some(0.0)
-                } else if self.null_on_divide_by_zero {
+            count if count == 1.0 && StatsType::Sample == self.stats_type => {
+                if self.null_on_divide_by_zero {
                     None
                 } else {
                     Some(f64::NAN)

spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala

@@ -958,9 +958,6 @@ class CometSparkSessionExtensions
           normalizePlan(plan)
         }
 
-        // scalastyle:off println
-        // println("isCometShuffleEnabled: " + isCometShuffleEnabled(conf))
-
         var newPlan = transform(normalizedPlan)
 
         // if the plan cannot be run fully natively then explain why (when appropriate
@@ -1189,15 +1186,9 @@ object CometSparkSessionExtensions extends Logging {
   // 1. `COMET_EXEC_SHUFFLE_ENABLED` is true
   // 2. `spark.shuffle.manager` is set to `CometShuffleManager`
   // 3. Off-heap memory is enabled || Spark/Comet unit testing
-  private[comet] def isCometShuffleEnabled(conf: SQLConf): Boolean = {
-    // scalastyle:off println
-    // println("COMET_EXEC_SHUFFLE_ENABLED: " + COMET_EXEC_SHUFFLE_ENABLED.get(conf))
-    // println("isCometShuffleManagerEnabled: " + isCometShuffleManagerEnabled(conf))
-    // println("isOffHeapEnabled: " + isOffHeapEnabled(conf))
-    // println("isTesting: " + isTesting)
+  private[comet] def isCometShuffleEnabled(conf: SQLConf): Boolean =
     COMET_EXEC_SHUFFLE_ENABLED.get(conf) && isCometShuffleManagerEnabled(conf) &&
-    (isOffHeapEnabled(conf) || isTesting)
-  }
+      (isOffHeapEnabled(conf) || isTesting)
 
   private[comet] def getCometShuffleNotEnabledReason(conf: SQLConf): Option[String] = {
     if (!COMET_EXEC_SHUFFLE_ENABLED.get(conf)) {