[XGBoost] update xgboost dependency to the 2.2.0-SNAPSHOT

Signed-off-by: Bobby Wang <[email protected]>
NVIDIA · Sep 25, 2024 · 942d3db · 942d3db
1 parent 3cca58b
commit 942d3db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/examples/XGBoost-Examples/pom.xml b/examples/XGBoost-Examples/pom.xml
@@ -38,18 +38,13 @@
 
     <properties>
         <encoding>UTF-8</encoding>
-        <xgboost.version>2.1.0-SNAPSHOT</xgboost.version>
+        <xgboost.version>2.2.0-SNAPSHOT</xgboost.version>
         <spark.version>3.5.0</spark.version>
         <scala.version>2.12.8</scala.version>
         <scala.binary.version>2.12</scala.binary.version>
     </properties>
 
     <dependencies>
-        <dependency>
-            <groupId>ml.dmlc</groupId>
-            <artifactId>xgboost4j-gpu_${scala.binary.version}</artifactId>
-            <version>${xgboost.version}</version>
-        </dependency>
         <dependency>
             <groupId>ml.dmlc</groupId>
             <artifactId>xgboost4j-spark-gpu_${scala.binary.version}</artifactId>

diff --git a/...es/XGBoost-Examples/utility/scala/src/com/nvidia/spark/examples/utility/XGBoostArgs.scala b/...es/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
   }