compiles

dillitz · Aug 9, 2024 · 7f9d34d · 7f9d34d
1 parent 37f215d
commit 7f9d34d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/spark-connect/client/src/main/scala-spark-master/io/delta/connect/tables/DeltaTable.scala b/spark-connect/client/src/main/scala-spark-master/io/delta/connect/tables/DeltaTable.scala
@@ -68,7 +68,8 @@ class DeltaTable private[tables](
       .newBuilder()
       .setTable(table)
     val relation = proto.DeltaRelation.newBuilder().setDescribeHistory(describeHistory).build()
-    val extension = org.sparkproject.com.google.protobuf.Any.pack(relation)
+    val extension = org.sparkproject.com.google.protobuf.Any
+      .pack(relation.asInstanceOf[org.sparkproject.com.google.protobuf.Message])
     val df = sparkSession.newDataFrame(_.setExtension(extension))
     limit match {
       case Some(limit) => df.limit(limit)
@@ -245,7 +246,7 @@ object DeltaTable {
       .newBuilder()
       .setScan(proto.Scan.newBuilder().setTable(table))
       .build()
-    val extension = org.sparkproject.com.google.protobuf.Any.pack(relation)
+    val extension = com.google.protobuf.Any.pack(relation)
     val sparkRelation = spark_proto.Relation.newBuilder().setExtension(extension).build()
     val df = sparkSession.newDataFrame(_.mergeFrom(sparkRelation))
     new DeltaTable(df, table)