Skip to content

Commit

Permalink
Updated ShimServiceProvider to return the correct Shim version
Browse files Browse the repository at this point in the history
  • Loading branch information
razajafri committed Sep 26, 2024
1 parent 5709d4a commit 8979789
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -24,14 +24,14 @@ import com.nvidia.spark.rapids.{DatabricksShimVersion, ShimVersion}
import org.apache.spark.SparkEnv

object SparkShimServiceProvider {
val VERSION = DatabricksShimVersion(3, 4, 1)
val VERSION = DatabricksShimVersion(3, 5, 0)
}

class SparkShimServiceProvider extends com.nvidia.spark.rapids.SparkShimServiceProvider {

override def getShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

def matchesVersion(version: String): Boolean = {
SparkEnv.get.conf.get("spark.databricks.clusterUsageTags.sparkVersion", "").startsWith("13.3.")
SparkEnv.get.conf.get("spark.databricks.clusterUsageTags.sparkVersion", "").startsWith("14.3.")
}
}

0 comments on commit 8979789

Please sign in to comment.