Fix decimal average signature

oap-project · Dec 19, 2023 · fcd23a5 · fcd23a5
1 parent 6ff0e99
commit fcd23a5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/velox/functions/sparksql/aggregates/AverageAggregate.cpp b/velox/functions/sparksql/aggregates/AverageAggregate.cpp
@@ -394,6 +394,15 @@ exec::AggregateRegistrationResult registerAverage(
           .returnType("DECIMAL(r_precision, r_scale)")
           .build());
 
+  signatures.push_back(
+      exec::AggregateFunctionSignatureBuilder()
+          .integerVariable("a_precision")
+          .integerVariable("a_scale")
+          .argumentType("DECIMAL(a_precision, a_scale)")
+          .intermediateType("ROW(DECIMAL(a_precision, a_scale), BIGINT)")
+          .returnType("DECIMAL(a_precision, a_scale)")
+          .build());
+
   return exec::registerAggregateFunction(
       name,
       std::move(signatures),