Fix build

Author: liurenjie1024

iceberg/src/main/scala/com/nvidia/spark/rapids/iceberg/GpuIcebergPartitioner.scala

@@ -20,7 +20,7 @@ import java.lang.Math.toIntExact
 
 import scala.collection.JavaConverters._
 
-import ai.rapids.cudf.{ColumnVector => CudfColumnVector, Table}
+import ai.rapids.cudf.{ColumnVector => CudfColumnVector, OrderByArg, Scalar, Table}
 import com.nvidia.spark.rapids.{GpuBoundReference, GpuColumnVector, GpuExpression, GpuLiteral, RapidsHostColumnVector, SpillableColumnarBatch, SpillPriorities}
 import com.nvidia.spark.rapids.Arm.{closeOnExcept, withResource}
 import com.nvidia.spark.rapids.RapidsPluginImplicits.AutoCloseableProducingSeq
@@ -54,12 +54,9 @@ class GpuIcebergPartitioner(val spec: PartitionSpec,
   private val partitionExprs: Seq[GpuExpression] = spec.fields().asScala.map(getPartitionExpr).toSeq
 
   private val keyColNum: Int = spec.fields().size()
-  private val inputColNum: Int = dataSparkType.fields.length
 
   // key column indices in the table: [key columns, input columns]
   private val keyColIndices: Array[Int] = (0 until keyColNum).toArray
-  // input column indices in the table: [key columns, input columns]
-  private val inputColumnIndices: Array[Int] = (keyColNum until (keyColNum + inputColNum)).toArray
 
   /**
    * Make a new table: [key columns, input columns]
@@ -109,7 +106,7 @@ class GpuIcebergPartitioner(val spec: PartitionSpec,
       // note: the result does not contain the key columns
       val splitRet = withResource(keysAndInputTable) { _ =>
         keysAndInputTable.groupBy(keyColIndices: _*)
-          .contiguousSplitGroupsAndGenUniqKeys(inputColumnIndices)
+          .contiguousSplitGroupsAndGenUniqKeys()
       }
 
       // generate results