diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/Exchange.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/Exchange.scala index 21029861709a6..e66768d109690 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/execution/Exchange.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/Exchange.scala @@ -269,7 +269,7 @@ private[sql] case class EnsureRequirements(sqlContext: SQLContext) extends Rule[ operator.withNewChildren(newChildren) } - def apply(plan: SparkPlan): SparkPlan = plan.transformUp { + def apply(plan: SparkPlan): SparkPlan = plan.transformUp { case operator: SparkPlan => ensureDistributionAndOrdering(ensureChildNumPartitionsAgreementIfNecessary(operator)) }