fixed the failing shim

NVIDIA · May 30, 2024 · beed117 · beed117
1 parent 565ca09
commit beed117
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/sql-plugin/src/main/spark350/scala/com/nvidia/spark/rapids/shims/BatchScanExecMeta.scala b/sql-plugin/src/main/spark350/scala/com/nvidia/spark/rapids/shims/BatchScanExecMeta.scala
@@ -22,8 +22,6 @@ package com.nvidia.spark.rapids.shims
 
 import com.nvidia.spark.rapids._
 
-import org.apache.spark.sql.catalyst.expressions._
-import org.apache.spark.sql.execution._
 import org.apache.spark.sql.execution.datasources.v2.BatchScanExec
 
 class BatchScanExecMeta(p: BatchScanExec,

diff --git a/sql-plugin/src/main/spark350/scala/com/nvidia/spark/rapids/shims/BatchScanExecMetaBase.scala b/sql-plugin/src/main/spark350/scala/com/nvidia/spark/rapids/shims/BatchScanExecMetaBase.scala
@@ -27,7 +27,7 @@ import org.apache.spark.sql.catalyst.expressions._
 import org.apache.spark.sql.execution._
 import org.apache.spark.sql.execution.datasources.v2.BatchScanExec
 
-class BatchScanExecMetaBase(p: BatchScanExec,
+abstract class BatchScanExecMetaBase(p: BatchScanExec,
    conf: RapidsConf,
    parent: Option[RapidsMeta[_, _, _]],
    rule: DataFromReplacementRule)
@@ -38,7 +38,7 @@ class BatchScanExecMetaBase(p: BatchScanExec,
   // BatchScanExec is independent from the replacement of the runtime filters. It is
   // possible that the BatchScanExec is on the CPU, while the dynamic runtime filters
   // are on the GPU. And vice versa.
-  private lazy val runtimeFilters = {
+  protected lazy val runtimeFilters = {
     val convertBroadcast = (bc: SubqueryBroadcastExec) => {
       val meta = GpuOverrides.wrapAndTagPlan(bc, conf)
       meta.tagForExplain()