From 7bf3931a462afb842d2ca9bd1550a5ce0e66a273 Mon Sep 17 00:00:00 2001
From: Bobby Wang <wbo4958@gmail.com>
Date: Wed, 25 Sep 2024 21:11:39 +0800
Subject: [PATCH] [XGBoost] update xgboost dependency to the 2.2.0-SNAPSHOT
 (#435)

Signed-off-by: Bobby Wang <wbo4958@gmail.com>
---
 examples/XGBoost-Examples/pom.xml                          | 7 +------
 .../com/nvidia/spark/examples/utility/XGBoostArgs.scala    | 4 +---
 2 files changed, 2 insertions(+), 9 deletions(-)

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 @@
 
     <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/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
   }