Fix Databricks building

NVIDIA · Jan 14, 2025 · 3d3b172 · 3d3b172
1 parent 99602c5
commit 3d3b172
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 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
@@ -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/.../src/main/spark320/scala/com/nvidia/spark/rapids/shims/HybridFileSourceScanExecMeta.scala b/.../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"}