fix comments

facebookincubator · Dec 16, 2023 · 8f68b66 · 8f68b66
1 parent f63dc38
commit 8f68b66
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/velox/functions/lib/aggregates/CentralMomentsAggregatesBase.h b/velox/functions/lib/aggregates/CentralMomentsAggregatesBase.h
@@ -32,7 +32,7 @@ struct CentralMomentsIndices {
 constexpr CentralMomentsIndices kCentralMomentsIndices{0, 1, 2, 3, 4};
 
 struct CentralMomentsAccumulator {
-  double count() const {
+  int64_t count() const {
     return count_;
   }
 
@@ -83,6 +83,14 @@ struct CentralMomentsAccumulator {
     if (otherCount == 0) {
       return;
     }
+    if (count_ == 0) {
+      count_ = otherCount;
+      m1_ = otherM1;
+      m2_ = otherM2;
+      m3_ = otherM3;
+      m4_ = otherM4;
+      return;
+    }
 
     double oldCount = count();
     count_ += otherCount;