Merge pull request dmlc#1010 from CodingCat/master

[jvm-packages] adjust numWorkers for test
DroLuo · Mar 18, 2016 · 11415f2 · 11415f2
2 parents a214670 + 14a031a
commit 11415f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSuite.scala b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSuite.scala
@@ -35,7 +35,7 @@ import ml.dmlc.xgboost4j.scala.{Booster, DMatrix, EvalTrait}
 class XGBoostSuite extends FunSuite with BeforeAndAfter {
 
   private implicit var sc: SparkContext = null
-  private val numWorkers = 4
+  private val numWorkers = Runtime.getRuntime().availableProcessors()
 
   private class EvalError extends EvalTrait {