update supportedDelta

apache · Sep 12, 2024 · 2cbf09a · 2cbf09a
1 parent 0065dd7
commit 2cbf09a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/gluten-substrait/src/main/scala/org/apache/gluten/execution/FilterTransformerFactory.scala b/gluten-substrait/src/main/scala/org/apache/gluten/execution/FilterTransformerFactory.scala
@@ -75,8 +75,13 @@ object FilterTransformerFactory {
   private def supportedDelta(plan: SparkPlan): Boolean = {
     plan.find {
       p =>
-        p.isInstanceOf[FileSourceScanExec] &&
-        p.asInstanceOf[FileSourceScanExec].relation.fileFormat.getClass.getName.contains("Delta")
+        p.isInstanceOf[
+          FileSourceScanExec] && "org.apache.spark.sql.delta.DeltaParquetFileFormat" == p
+          .asInstanceOf[FileSourceScanExec]
+          .relation
+          .fileFormat
+          .getClass
+          .getName
     }.isDefined
   }
 }
diff --git a/gluten-substrait/src/main/scala/org/apache/gluten/execution/ProjectTransformerFactory.scala b/gluten-substrait/src/main/scala/org/apache/gluten/execution/ProjectTransformerFactory.scala
@@ -75,8 +75,13 @@ object ProjectTransformerFactory {
   private def supportedDelta(plan: SparkPlan): Boolean = {
     plan.find {
       p =>
-        p.isInstanceOf[FileSourceScanExec] &&
-        p.asInstanceOf[FileSourceScanExec].relation.fileFormat.getClass.getName.contains("Delta")
+        p.isInstanceOf[
+          FileSourceScanExec] && "org.apache.spark.sql.delta.DeltaParquetFileFormat" == p
+          .asInstanceOf[FileSourceScanExec]
+          .relation
+          .fileFormat
+          .getClass
+          .getName
     }.isDefined
   }
 }