From a1b875e32f9e2c0b7c66c4bedda3f9aad121af6c Mon Sep 17 00:00:00 2001
From: hpal <hpal@apple.com>
Date: Tue, 27 Feb 2024 23:52:08 -0800
Subject: [PATCH] revert to previous partition spec

---
 .../spark/sql/connector/KeyGroupedPartitioningSuite.scala       | 2 +-
 1 file changed, 1 insertion(+), 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
index adeb1d5413116..a06a462e6ef0b 100644
--- 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
@@ -1484,7 +1484,7 @@ class KeyGroupedPartitioningSuite extends DistributionAndOrderingSuiteBase {
 
         val partition1 = Array(identity("store_id"),
           bucket(table1buckets, "dept_id"))
-        val partition2 = Array(identity("store_id"),
+        val partition2 = Array(bucket(3, "store_id"),
           bucket(table2buckets, "dept_id"))
 
         createTable(table1, schema2, partition1)