[jvm-packages] fix group col for gpu packages

wbo4958 · May 8, 2024 · 295a95f · 295a95f
1 parent 8237920
commit 295a95f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...boost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala b/...boost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala
@@ -160,7 +160,7 @@ object GpuPreXGBoost extends PreXGBoostProvider {
 
     // Check columns and build column data batch
     val trainingData = GpuUtils.buildColumnDataBatch(feturesCols,
-      labelName, weightName, marginName, "", castedDF)
+      labelName, weightName, marginName, groupName, castedDF)
 
     // eval map
     val evalDataMap = evalSets.map {