fix(spark): handle AggregationPhase UNSPECIFIED

substrait-io · Dec 24, 2024 · bedd309 · bedd309
1 parent 116f1a7
commit bedd309
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spark/src/main/scala/io/substrait/spark/expression/ToAggregateFunction.scala b/spark/src/main/scala/io/substrait/spark/expression/ToAggregateFunction.scala
@@ -74,6 +74,8 @@ object ToAggregateFunction {
     case other => throw new UnsupportedOperationException(s"not currently supported: $other.")
   }
   def toSpark(phase: SExpression.AggregationPhase): AggregateMode = phase match {
+    case SExpression.AggregationPhase.UNSPECIFIED =>
+      Final // UNSPECIFIED implies INTERMEDIATE_TO_RESULT
     case SExpression.AggregationPhase.INITIAL_TO_INTERMEDIATE => Partial
     case SExpression.AggregationPhase.INTERMEDIATE_TO_INTERMEDIATE => PartialMerge
     case SExpression.AggregationPhase.INTERMEDIATE_TO_RESULT => Final