From 8f6aadd4b7e50181af920660fe0c4db3a831dd3a Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Sun, 20 May 2018 03:28:11 +0200 Subject: [PATCH] [jvm-packages] Fixed CheckpointManagerSuite for Scala 2.10 (#3332) As before, the compilation error is caused by mixing positional and labelled arguments. --- .../dmlc/xgboost4j/scala/spark/CheckpointManagerSuite.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/CheckpointManagerSuite.scala b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/CheckpointManagerSuite.scala index 4a6feb2e4..af012c4f5 100644 --- a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/CheckpointManagerSuite.scala +++ b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/CheckpointManagerSuite.scala @@ -38,8 +38,8 @@ class CheckpointManagerSuite extends FunSuite with BeforeAndAfterAll { val trainingRDD = sc.parallelize(Classification.train).map(_.asML).cache() val paramMap = Map("eta" -> "1", "max_depth" -> "2", "silent" -> "1", "objective" -> "binary:logistic") - (XGBoost.trainWithRDD(trainingRDD, paramMap, round = 2, sc.defaultParallelism), - XGBoost.trainWithRDD(trainingRDD, paramMap, round = 4, sc.defaultParallelism)) + (XGBoost.trainWithRDD(trainingRDD, paramMap, round = 2, nWorkers = sc.defaultParallelism), + XGBoost.trainWithRDD(trainingRDD, paramMap, round = 4, nWorkers = sc.defaultParallelism)) } test("test update/load models") {