fix tests failure

vanzin · May 1, 2015 · e230155 · e230155
1 parent b09b895
commit e230155
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/Optimizer.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/Optimizer.scala
@@ -175,7 +175,7 @@ object ColumnPruning extends Rule[LogicalPlan] {
 
     // push down project if possible when the child is sort
     case p @ Project(projectList, s @ Sort(order, _, grandChild))
-      if (s.references -- p.references).size == 0 =>
+      if (s.references -- p.outputSet).isEmpty =>
       s.copy(child = Project(projectList, grandChild))
 
     // Eliminate no-op Projects