From 29b5b09f993ddb66fdb7e1cb1ab13123e63da928 Mon Sep 17 00:00:00 2001
From: Axel Pettersson <axel@pettersson.cc>
Date: Thu, 5 May 2022 15:34:51 +0200
Subject: [PATCH] Fix partitioning distribution of data when no equiality join

---
 scala/src/main/scala/execution/PITJoinExec.scala | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scala/src/main/scala/execution/PITJoinExec.scala b/scala/src/main/scala/execution/PITJoinExec.scala
index f9423f7..c3f14cc 100644
--- a/scala/src/main/scala/execution/PITJoinExec.scala
+++ b/scala/src/main/scala/execution/PITJoinExec.scala
@@ -72,7 +72,8 @@ protected[pit] case class PITJoinExec(
 
   override def requiredChildDistribution: Seq[Distribution] = {
     if (leftEquiKeys.isEmpty || rightEquiKeys.isEmpty) {
-      UnspecifiedDistribution :: UnspecifiedDistribution :: Nil
+      // TODO: This should be improved, but for now just keep everything in one partition
+      AllTuples :: AllTuples :: Nil
     } else {
       HashClusteredDistribution(leftEquiKeys) :: HashClusteredDistribution(
         rightEquiKeys