[jvm-packages] fix the split of input (#4417)

dmlc · Apr 30, 2019 · 253fdd8 · 253fdd8
1 parent 91c513a
commit 253fdd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...es/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostGeneralSuite.scala b/...es/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostGeneralSuite.scala
@@ -329,7 +329,7 @@ class XGBoostGeneralSuite extends FunSuite with PerTest {
 
   test("train with multiple validation datasets (ranking)") {
     val training = buildDataFrameWithGroup(Ranking.train, 5)
-    val Array(train, eval1, eval2) = training.randomSplit(Array(0.6, 0.2, 0.2))
+    val Array(train, eval1, eval2) = training.randomSplit(Array(0.6, 0.2, 0.2), 0)
     val paramMap1 = Map("eta" -> "1", "max_depth" -> "6",
       "objective" -> "rank:pairwise",
       "num_round" -> 5, "num_workers" -> numWorkers, "group_col" -> "group")