Skip to content

Commit 77fb305

Browse files
authored
Merge pull request #436 from NVIDIA/branch-24.10
[auto-merge] branch-24.10 to branch-24.12 [skip ci] [bot]
2 parents 6b1c689 + 7bf3931 commit 77fb305

File tree

2 files changed

+2
-9
lines changed

2 files changed

+2
-9
lines changed

examples/XGBoost-Examples/pom.xml

+1-6
Original file line numberDiff line numberDiff line change
@@ -38,18 +38,13 @@
3838

3939
<properties>
4040
<encoding>UTF-8</encoding>
41-
<xgboost.version>2.1.0-SNAPSHOT</xgboost.version>
41+
<xgboost.version>2.2.0-SNAPSHOT</xgboost.version>
4242
<spark.version>3.5.0</spark.version>
4343
<scala.version>2.12.8</scala.version>
4444
<scala.binary.version>2.12</scala.binary.version>
4545
</properties>
4646

4747
<dependencies>
48-
<dependency>
49-
<groupId>ml.dmlc</groupId>
50-
<artifactId>xgboost4j-gpu_${scala.binary.version}</artifactId>
51-
<version>${xgboost.version}</version>
52-
</dependency>
5348
<dependency>
5449
<groupId>ml.dmlc</groupId>
5550
<artifactId>xgboost4j-spark-gpu_${scala.binary.version}</artifactId>

examples/XGBoost-Examples/utility/scala/src/com/nvidia/spark/examples/utility/XGBoostArgs.scala

+1-3
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,6 @@ import com.google.common.base.CaseFormat
2222
import scala.collection.mutable
2323
import scala.util.Try
2424

25-
import ml.dmlc.xgboost4j.scala.spark.TrackerConf
26-
2725
private case class XGBoostArg(
2826
required: Boolean = false,
2927
parse: String => Any = value => value,
@@ -202,7 +200,7 @@ class XGBoostArgs private[utility] (
202200

203201
val hostIp = params.getOrElse("rabit_tracker_host", "").toString
204202
if (!hostIp.isEmpty) {
205-
params ++ Map("tracker_conf" -> TrackerConf(0, hostIp))
203+
params ++ Map("rabitTrackerHostIp" -> hostIp)
206204
} else params
207205
}
208206

0 commit comments

Comments
 (0)