diff --git a/examples/XGBoost-Examples/pom.xml b/examples/XGBoost-Examples/pom.xml index 22b1a1315..d23c57e49 100644 --- a/examples/XGBoost-Examples/pom.xml +++ b/examples/XGBoost-Examples/pom.xml @@ -38,18 +38,13 @@ UTF-8 - 2.1.0-SNAPSHOT + 2.2.0-SNAPSHOT 3.5.0 2.12.8 2.12 - - ml.dmlc - xgboost4j-gpu_${scala.binary.version} - ${xgboost.version} - ml.dmlc xgboost4j-spark-gpu_${scala.binary.version} diff --git a/examples/XGBoost-Examples/utility/scala/src/com/nvidia/spark/examples/utility/XGBoostArgs.scala b/examples/XGBoost-Examples/utility/scala/src/com/nvidia/spark/examples/utility/XGBoostArgs.scala index d58e8e94b..2e8cf0fc8 100644 --- a/examples/XGBoost-Examples/utility/scala/src/com/nvidia/spark/examples/utility/XGBoostArgs.scala +++ b/examples/XGBoost-Examples/utility/scala/src/com/nvidia/spark/examples/utility/XGBoostArgs.scala @@ -22,8 +22,6 @@ import com.google.common.base.CaseFormat import scala.collection.mutable import scala.util.Try -import ml.dmlc.xgboost4j.scala.spark.TrackerConf - private case class XGBoostArg( required: Boolean = false, parse: String => Any = value => value, @@ -202,7 +200,7 @@ class XGBoostArgs private[utility] ( val hostIp = params.getOrElse("rabit_tracker_host", "").toString if (!hostIp.isEmpty) { - params ++ Map("tracker_conf" -> TrackerConf(0, hostIp)) + params ++ Map("rabitTrackerHostIp" -> hostIp) } else params }