[jvm-packages] refine tracker (#10313)

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-05-23 12:46:21 +08:00
parent 966dc81788
commit 932d7201f9
8 changed files with 71 additions and 92 deletions
--- a/jvm-packages/xgboost4j-flink/src/main/java/ml/dmlc/xgboost4j/java/flink/XGBoost.java
+++ b/jvm-packages/xgboost4j-flink/src/main/java/ml/dmlc/xgboost4j/java/flink/XGBoost.java
@@ -176,7 +176,7 @@ public class XGBoost {
        new RabitTracker(dtrain.getExecutionEnvironment().getParallelism());
    if (tracker.start()) {
      return dtrain
-        .mapPartition(new MapFunction(params, numBoostRound, tracker.workerArgs()))
+        .mapPartition(new MapFunction(params, numBoostRound, tracker.getWorkerArgs()))
        .reduce((x, y) -> x)
        .collect()
        .get(0);