diff --git a/sql-plugin/src/main/java/com/nvidia/spark/rapids/UnsafeRowToColumnarBatchIterator.java b/sql-plugin/src/main/java/com/nvidia/spark/rapids/UnsafeRowToColumnarBatchIterator.java index 8f35fc037ba..627d6a264c7 100644 --- a/sql-plugin/src/main/java/com/nvidia/spark/rapids/UnsafeRowToColumnarBatchIterator.java +++ b/sql-plugin/src/main/java/com/nvidia/spark/rapids/UnsafeRowToColumnarBatchIterator.java @@ -49,7 +49,7 @@ public abstract class UnsafeRowToColumnarBatchIterator implements Iterator outputTypes; + protected final DataType[] outputTypes; protected final SQLMetric totalTime; protected final SQLMetric numInputRows; protected final SQLMetric numOutputRows; @@ -69,11 +69,11 @@ protected UnsafeRowToColumnarBatchIterator( Math.min(Integer.MAX_VALUE - 1, goal.targetSizeBytes() / sizePerRowEstimate)); dataLength = ((long) sizePerRowEstimate) * numRowsEstimate; rapidsTypes = new DType[schema.length]; - outputTypes = new ArrayList<>(schema.length); + outputTypes = new DataType[schema.length]; for (int i = 0; i < schema.length; i++) { rapidsTypes[i] = GpuColumnVector.getRapidsType(schema[i].dataType()); - outputTypes.add(schema[i].dataType()); + outputTypes[i] = schema[i].dataType(); } this.totalTime = totalTime; this.numInputRows = numInputRows;