From 8066d6d3ec8a005754c1d8eedb7b316c6ae8ca3b Mon Sep 17 00:00:00 2001 From: ashahid Date: Wed, 15 Jan 2025 08:04:04 -0800 Subject: [PATCH] SPARK-49881. fixed compilation failure after merge --- .../main/scala/org/apache/spark/sql/connect/ml/MLUtils.scala | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sql/connect/server/src/main/scala/org/apache/spark/sql/connect/ml/MLUtils.scala b/sql/connect/server/src/main/scala/org/apache/spark/sql/connect/ml/MLUtils.scala index 72c86401eb631..f285d286e5426 100644 --- a/sql/connect/server/src/main/scala/org/apache/spark/sql/connect/ml/MLUtils.scala +++ b/sql/connect/server/src/main/scala/org/apache/spark/sql/connect/ml/MLUtils.scala @@ -30,6 +30,7 @@ import org.apache.spark.ml.linalg.{Matrices, Matrix, Vector, Vectors} import org.apache.spark.ml.param.Params import org.apache.spark.ml.util.MLWritable import org.apache.spark.sql.{DataFrame, Dataset} +import org.apache.spark.sql.catalyst.analysis.RelationWrapper import org.apache.spark.sql.connect.common.LiteralValueProtoConverter import org.apache.spark.sql.connect.planner.SparkConnectPlanner import org.apache.spark.sql.connect.service.SessionHolder @@ -188,6 +189,7 @@ private[ml] object MLUtils { def parseRelationProto(relation: proto.Relation, sessionHolder: SessionHolder): DataFrame = { val planner = new SparkConnectPlanner(sessionHolder) val plan = planner.transformRelation(relation) + implicit val withRelations: Set[RelationWrapper] = Set.empty Dataset.ofRows(sessionHolder.session, plan) }