Fix checkConnectJvmClientCompatibility

szehon-ho · Aug 6, 2024 · 0fb2841 · 0fb2841
1 parent a77f15b
commit 0fb2841
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...c/test/scala/org/apache/spark/sql/connect/client/CheckConnectJvmClientCompatibility.scala b/...c/test/scala/org/apache/spark/sql/connect/client/CheckConnectJvmClientCompatibility.scala
@@ -318,9 +318,13 @@ object CheckConnectJvmClientCompatibility {
         "org.apache.spark.sql.artifact.util.ArtifactUtils$"),
 
       // Update Writer
+      ProblemFilters.exclude[MissingMethodProblem]("org.apache.spark.sql.SparkSession.update"),
       ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWriter"),
+      ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWriter$"),
       ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWithAssignment"),
-      ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWithCondition")) ++
+      ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWithAssignment$"),
+      ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWithCondition"),
+      ProblemFilters.exclude[MissingClassProblem]("org.apache.spark.sql.UpdateWithCondition$")) ++
       mergeIntoWriterExcludeRules
 
     checkMiMaCompatibility(clientJar, sqlJar, includedRules, excludeRules)