SPARK-49881. fixed compilation failure after merge

ahshahid · Jan 15, 2025 · 8066d6d · 8066d6d
1 parent c14bb66
commit 8066d6d
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -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)
   }