fix db test failures

Signed-off-by: Raza Jafri <rjafri@nvidia.com>
razajafri · Mar 9, 2022 · 7782695 · 7782695
1 parent ac2c449
commit 7782695
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/sql-plugin/src/main/301db/scala/com/nvidia/spark/rapids/SparkShims.scala b/sql-plugin/src/main/301db/scala/com/nvidia/spark/rapids/SparkShims.scala
@@ -17,8 +17,9 @@
 package com.nvidia.spark.rapids.shims
 
 import com.nvidia.spark.rapids._
+import com.nvidia.spark.rapids.shims.spark301db.SparkShimServiceProvider
 
 object SparkShimImpl extends Spark30XdbShims with Spark30Xuntil33XShims {
 
-  override def getSparkShimVersion: ShimVersion = ShimLoader.getShimVersion
+  override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION
 }
diff --git a/sql-plugin/src/main/312db/scala/com/nvidia/spark/rapids/shims/SparkShims.scala b/sql-plugin/src/main/312db/scala/com/nvidia/spark/rapids/shims/SparkShims.scala
@@ -17,14 +17,15 @@
 package com.nvidia.spark.rapids.shims
 
 import com.nvidia.spark.rapids._
+import com.nvidia.spark.rapids.shims.spark312db.SparkShimServiceProvider
 import org.apache.parquet.schema.MessageType
 
 import org.apache.spark.sql.execution.datasources.DataSourceUtils
 import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
 
 object SparkShimImpl extends Spark31XdbShims with Spark30Xuntil33XShims {
 
-  override def getSparkShimVersion: ShimVersion = ShimLoader.getShimVersion
+  override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION
 
   override def getParquetFilters(
       schema: MessageType,