Github user sujith71955 commented on a diff in the pull request:
https://github.com/apache/spark/pull/16677#discussion_r204362254
--- Diff:
sql/core/src/main/scala/org/apache/spark/sql/execution/exchange/ShuffleExchangeExec.scala
---
@@ -231,6 +231,12 @@ object ShuffleExchangeExec {
override def numPartitions: Int = 1
override def getPartition(key: Any): Int = 0
}
+ case l: LocalPartitioning =>
+ new Partitioner {
+ override def numPartitions: Int = l.numPartitions
+ override def getPartition(key: Any): Int = key.asInstanceOf[Int]
+ }
+
--- End diff --
nit :extra space
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
