fix

Author: zhengruifeng

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/plans/QueryPlan.scala

@@ -66,6 +66,7 @@ abstract class QueryPlan[PlanType <: QueryPlan[PlanType]]
    */
   def outputOrdering: Seq[SortOrder] = Nil
 
+  // Override `treePatternBits` to propagate bits for its expressions.
   private val _treePatternBits = new BestEffortLazyVal[BitSet](() => {
     val bits: BitSet = getDefaultTreePatternBits
     // Propagate expressions' pattern bits
@@ -75,8 +76,6 @@ abstract class QueryPlan[PlanType <: QueryPlan[PlanType]]
     }
     bits
   })
-
-  // Override `treePatternBits` to propagate bits for its expressions.
   override def treePatternBits: BitSet = _treePatternBits()
 
   /**

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/trees/TreeNode.scala

@@ -108,12 +108,11 @@ abstract class TreeNode[BaseType <: TreeNode[BaseType]]
     bits
   }
 
-  private val _treePatternBits = new BestEffortLazyVal[BitSet](() => getDefaultTreePatternBits)
-
   /**
    * A BitSet of tree patterns for this TreeNode and its subtree. If this TreeNode and its
    * subtree contains a pattern `P`, the corresponding bit for `P.id` is set in this BitSet.
    */
+  private val _treePatternBits = new BestEffortLazyVal[BitSet](() => getDefaultTreePatternBits)
   override def treePatternBits: BitSet = _treePatternBits()
 
   /**