spark311db build fix

NVIDIA · Sep 10, 2021 · f01b0b6 · f01b0b6
1 parent 8dcf0e1
commit f01b0b6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...s/spark311db/src/main/scala/com/nvidia/spark/rapids/shims/spark311db/SparkBaseShims.scala b/...s/spark311db/src/main/scala/com/nvidia/spark/rapids/shims/spark311db/SparkBaseShims.scala
@@ -21,6 +21,7 @@ import java.nio.ByteBuffer
 
 import com.databricks.sql.execution.window.RunningWindowFunctionExec
 import com.nvidia.spark.rapids._
+import com.nvidia.spark.rapids.shims.spark311db._
 import com.nvidia.spark.rapids.shims.v2.Spark30XShims
 import org.apache.arrow.memory.ReferenceManager
 import org.apache.arrow.vector.ValueVector
@@ -56,6 +57,7 @@ import org.apache.spark.sql.internal.{SQLConf, StaticSQLConf}
 import org.apache.spark.sql.rapids._
 import org.apache.spark.sql.rapids.execution.{GpuBroadcastExchangeExecBase, GpuBroadcastNestedLoopJoinExecBase, GpuShuffleExchangeExecBase, JoinTypeChecks}
 import org.apache.spark.sql.rapids.execution.python.{GpuAggregateInPandasExecMeta, GpuArrowEvalPythonExec, GpuMapInPandasExecMeta, GpuPythonUDF}
+import org.apache.spark.sql.rapids.execution.python.shims.spark311db._
 import org.apache.spark.sql.rapids.shims.spark311db._
 import org.apache.spark.sql.sources.BaseRelation
 import org.apache.spark.sql.types._