fix bug that numUsers * numProducts will exceed the manxmun value of …

…int. (Intel-bigdata#658) Signed-off-by: minmingzhu <[email protected]>
henrque · Nov 22, 2021 · 0768463 · 0768463
1 parent 32472f2
commit 0768463
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparkbench/ml/src/main/scala/com/intel/sparkbench/ml/RatingDataGenerator.scala b/sparkbench/ml/src/main/scala/com/intel/sparkbench/ml/RatingDataGenerator.scala
@@ -43,7 +43,7 @@ object RatingDataGenerator {
       numRatings = args(3).toInt
       implicitPrefs = args(4).toBoolean
 
-      require(numRatings <= numUsers * numProducts,
+      require(numRatings.toLong <= numUsers.toLong * numProducts.toLong,
         s"RatingDataGenerator: numRatings=$numRatings is too large, should be <= numUsers * numProducts"
       )