address review comments

kazuyukitanimura · May 22, 2024 · d45ac85 · d45ac85
1 parent 0b81d7f
commit d45ac85
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.../src/main/spark-3.x/org/apache/spark/sql/comet/shims/ShimCometShuffleWriteProcessor.scala b/.../src/main/spark-3.x/org/apache/spark/sql/comet/shims/ShimCometShuffleWriteProcessor.scala
@@ -20,10 +20,9 @@
 package org.apache.spark.sql.comet.shims
 
 import org.apache.spark.{Partition, ShuffleDependency, TaskContext}
-import org.apache.spark.rdd.{MapPartitionsRDD, RDD}
+import org.apache.spark.rdd.RDD
 import org.apache.spark.scheduler.MapStatus
 import org.apache.spark.shuffle.ShuffleWriteProcessor
-import org.apache.spark.sql.vectorized.ColumnarBatch
 
 trait ShimCometShuffleWriteProcessor extends ShuffleWriteProcessor {
   override def write(