minor change

himadripal · Feb 27, 2024 · 8db908f · 8db908f
1 parent dceea13
commit 8db908f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/connector/KeyGroupedPartitioningSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/connector/KeyGroupedPartitioningSuite.scala
@@ -1522,7 +1522,7 @@ class KeyGroupedPartitioningSuite extends DistributionAndOrderingSuiteBase {
           val scans = collectScans(df.queryExecution.executedPlan).map(_.inputRDD.
             partitions.length)
 
-          val expectedBuckets = Math.min(2, 4)
+          val expectedBuckets = Math.min(table1buckets, table2buckets)
 
           assert(scans == Seq(expectedBuckets, expectedBuckets))
           // Not coming till point