From a8bb98e0bbc66e62a8766157412df7df6b6dd020 Mon Sep 17 00:00:00 2001 From: Maomao Min Date: Wed, 20 Sep 2023 12:53:37 +0800 Subject: [PATCH] Update RowEncoder's usage for Spark 3.5.0 support Signed-off-by: Maomao Min --- .../scala/ml/dmlc/xgboost4j/scala/spark/rapids/GpuXGBoost.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/rapids/GpuXGBoost.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/rapids/GpuXGBoost.scala index cea13cd075b2..4588d2226079 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/rapids/GpuXGBoost.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/rapids/GpuXGBoost.scala @@ -380,7 +380,7 @@ private[spark] object GpuXGBoost { .groupBy(groupName) .agg(collect_list(struct(schema.fieldNames.map(col): _*)) as "list") - implicit val encoder = RowEncoder(schema) + implicit val encoder = RowEncoder.encoderFor(schema) // Expand the grouped rows after repartition groupedDF.repartition(nWorkers).mapPartitions(iter => { new Iterator[Row] {