fix

7mming7 · Jul 9, 2024 · 545baff · 545baff
1 parent 34a8b8c
commit 545baff
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/spark/src/main/scala/org/apache/spark/sql/delta/perf/OptimizeMetadataOnlyDeltaQuery.scala b/spark/src/main/scala/org/apache/spark/sql/delta/perf/OptimizeMetadataOnlyDeltaQuery.scala
@@ -134,9 +134,11 @@ trait OptimizeMetadataOnlyDeltaQuery extends Logging {
     }
   }
 
-  private def extractPartitionFilters(plan: Aggregate): Option[Expression] = {
-    plan.child match {
+  private def extractPartitionFilters(plan: LogicalPlan): Option[Expression] = {
+    plan match {
       case Filter(cond, _) => Some(cond)
+      case Project(_, child) => extractPartitionFilters(child)
+      case Aggregate(_, _, child) => extractPartitionFilters(child)
       case _ => None
     }
   }