revert: remove incorrect override of Project output schema

Reverts 77e7f8f

Author: ZorinAnton

core/src/main/java/io/substrait/relation/ProtoRelConverter.java

@@ -620,19 +620,8 @@ protected Expand newExpand(ExpandRel rel) {
 
   protected Aggregate newAggregate(AggregateRel rel) {
     Rel input = from(rel.getInput());
-    Type.Struct inputSchema;
-    if (input instanceof Project) {
-      List<Type> types =
-          ((Project) input)
-              .getExpressions().stream().map(Expression::getType).collect(Collectors.toList());
-      inputSchema = Type.Struct.builder().fields(types).nullable(false).build();
-    } else {
-      inputSchema = input.getRecordType();
-    }
-
     ProtoExpressionConverter protoExprConverter =
-        new ProtoExpressionConverter(lookup, extensions, inputSchema, this);
-
+        new ProtoExpressionConverter(lookup, extensions, input.getRecordType(), this);
     ProtoAggregateFunctionConverter protoAggrFuncConverter =
         new ProtoAggregateFunctionConverter(lookup, extensions, protoExprConverter);
 

isthmus/src/main/java/io/substrait/isthmus/SubstraitRelVisitor.java

@@ -244,60 +244,9 @@ public Rel visit(org.apache.calcite.rel.core.Minus minus) {
     return Set.builder().inputs(inputs).setOp(setOp).build();
   }
 
-  /**
-   * Pre-processes the input to an Aggregate relation to handle nullability changes introduced by
-   * ROLLUP/CUBE/GROUPING SETS.
-   *
-   * @param aggregate The original Calcite aggregate node.
-   * @return A Substrait Rel node that is correctly typed to be the input to the Substrait
-   *     Aggregate.
-   */
-  private Rel handleRollupCorrection(org.apache.calcite.rel.core.Aggregate aggregate) {
-    Rel originalInput = apply(aggregate.getInput());
-
-    // Determine the correct final output type for the aggregate, which accounts for nullability.
-    NamedStruct aggregateOutputType = typeConverter.toNamedStruct(aggregate.getRowType());
-    List<Integer> groupKeyIndices = aggregate.getGroupSet().asList();
-
-    // Create a list of expressions to cast the original input to the correct final type if needed.
-    List<Expression> castExpressions = new ArrayList<>();
-
-    boolean needsCasting = false;
-    for (int i = 0; i < originalInput.getRecordType().fields().size(); i++) {
-      Expression fieldReference = FieldReference.newInputRelReference(i, originalInput);
-
-      if (groupKeyIndices.contains(i)) {
-        int groupKeyOutputIndex = groupKeyIndices.indexOf(i);
-        Type finalType = aggregateOutputType.struct().fields().get(groupKeyOutputIndex);
-
-        if (finalType.nullable() && !fieldReference.getType().nullable()) {
-          needsCasting = true; // Mark that a cast is necessary.
-          castExpressions.add(
-              Expression.Cast.builder()
-                  .type(finalType)
-                  .input(fieldReference)
-                  .failureBehavior(Expression.FailureBehavior.RETURN_NULL)
-                  .build());
-        } else {
-          castExpressions.add(fieldReference);
-        }
-      } else {
-        castExpressions.add(fieldReference);
-      }
-    }
-
-    // Only add the extra Project node if a cast was actually needed.
-    if (needsCasting) {
-      return Project.builder().input(originalInput).expressions(castExpressions).build();
-    }
-
-    // If no casting was needed, just return the original converted input.
-    return originalInput;
-  }
-
   @Override
   public Rel visit(org.apache.calcite.rel.core.Aggregate aggregate) {
-    Rel input = handleRollupCorrection(aggregate);
+    Rel input = apply(aggregate.getInput());
     Stream<ImmutableBitSet> sets;
     if (aggregate.groupSets != null) {
       sets = aggregate.groupSets.stream();

isthmus/src/test/java/io/substrait/isthmus/TpcdsQueryTest.java

@@ -15,7 +15,7 @@
 public class TpcdsQueryTest extends PlanTestBase {
   private static final Set<Integer> toSubstraitExclusions = Set.of(9, 27, 36, 70, 86);
   private static final Set<Integer> fromSubstraitPojoExclusions = Set.of(1, 30, 81);
-  private static final Set<Integer> fromSubstraitProtoExclusions = Set.of(1, 30, 81);
+  private static final Set<Integer> fromSubstraitProtoExclusions = Set.of(1, 30, 67, 81);
 
   static IntStream testCases() {
     return IntStream.rangeClosed(1, 99).filter(n -> !toSubstraitExclusions.contains(n));