Updated datatype for count - StatisticsMetrics

Signed-off-by: Sayed Bilal Bari <sbari@nvidia.com>
NVIDIA · Feb 21, 2025 · 25aed33 · 25aed33
1 parent c3537e8
commit 25aed33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/scala/com/nvidia/spark/rapids/tool/analysis/StatisticsMetrics.scala b/core/src/main/scala/com/nvidia/spark/rapids/tool/analysis/StatisticsMetrics.scala
@@ -19,12 +19,12 @@ package com.nvidia.spark.rapids.tool.analysis
 import org.apache.spark.sql.rapids.tool.util.InPlaceMedianArrView.{chooseMidpointPivotInPlace, findMedianInPlace}
 
 // Store (min, median, max, total) for a given metric
-case class StatisticsMetrics(min: Long, med: Long, max: Long, count: Int, total: Long)
+case class StatisticsMetrics(min: Long, med: Long, max: Long, count: Long, total: Long)
 
 object StatisticsMetrics {
   // a static variable used to represent zero-statistics instead of allocating a dummy record
   // on every calculation.
-  val ZERO_RECORD: StatisticsMetrics = StatisticsMetrics(0L, 0L, 0L, 0, 0L)
+  val ZERO_RECORD: StatisticsMetrics = StatisticsMetrics(0L, 0L, 0L, 0L, 0L)
 
   def createFromArr(arr: Array[Long]): StatisticsMetrics = {
     if (arr.isEmpty) {