From 3d3b172374028a23058648b45dc98243794d39c3 Mon Sep 17 00:00:00 2001
From: Chong Gao <res_life@163.com>
Date: Tue, 14 Jan 2025 15:35:27 +0800
Subject: [PATCH] Fix Databricks building

---
 .../apache/spark/rapids/hybrid/HybridFileSourceScanExec.scala | 4 ++--
 .../spark/rapids/shims/HybridFileSourceScanExecMeta.scala     | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/sql-plugin/src/main/scala/org/apache/spark/rapids/hybrid/HybridFileSourceScanExec.scala b/sql-plugin/src/main/scala/org/apache/spark/rapids/hybrid/HybridFileSourceScanExec.scala
index 909a4eb3b80..37ba6996805 100644
--- a/sql-plugin/src/main/scala/org/apache/spark/rapids/hybrid/HybridFileSourceScanExec.scala
+++ b/sql-plugin/src/main/scala/org/apache/spark/rapids/hybrid/HybridFileSourceScanExec.scala
@@ -102,11 +102,11 @@ case class HybridFileSourceScanExec(originPlan: FileSourceScanExec
     inputRDD :: Nil
   }
 
-  private val commonMetrics = Map[String, () => GpuMetric](
+  private val commonMetrics: Map[String, () => GpuMetric] = Map[String, () => GpuMetric](
     "HybridScanTime" -> (() => createNanoTimingMetric(MODERATE_LEVEL, "HybridScanTime")),
     "GpuAcquireTime" -> (() => createNanoTimingMetric(MODERATE_LEVEL, "GpuAcquireTime")),
   )
-  private val nativeMetrics = Map[String, () => GpuMetric](
+  private val nativeMetrics: Map[String, () => GpuMetric] = Map[String, () => GpuMetric](
     "CoalescedBatches" -> (() => createMetric(MODERATE_LEVEL, "CoalescedBatches")),
     "CpuReaderBatches" -> (() => createMetric(MODERATE_LEVEL, "CpuReaderBatches")),
     "C2COutputSize" -> (() => createSizeMetric(MODERATE_LEVEL, "C2COutputSize")),
diff --git a/sql-plugin/src/main/spark320/scala/com/nvidia/spark/rapids/shims/HybridFileSourceScanExecMeta.scala b/sql-plugin/src/main/spark320/scala/com/nvidia/spark/rapids/shims/HybridFileSourceScanExecMeta.scala
index 9fb385ed4d1..5adc21ae7cb 100644
--- a/sql-plugin/src/main/spark320/scala/com/nvidia/spark/rapids/shims/HybridFileSourceScanExecMeta.scala
+++ b/sql-plugin/src/main/spark320/scala/com/nvidia/spark/rapids/shims/HybridFileSourceScanExecMeta.scala
@@ -23,17 +23,21 @@
 {"spark": "324"}
 {"spark": "330"}
 {"spark": "330cdh"}
+ {"spark": "330db"}
 {"spark": "331"}
 {"spark": "332"}
 {"spark": "332cdh"}
+{"spark": "332db"}
 {"spark": "333"}
 {"spark": "334"}
 {"spark": "340"}
 {"spark": "341"}
+{"spark": "341db"}
 {"spark": "342"}
 {"spark": "343"}
 {"spark": "344"}
 {"spark": "350"}
+{"spark": "350db143"}
 {"spark": "351"}
 {"spark": "352"}
 {"spark": "353"}