Force old default for group_keys

apache · Sep 20, 2023 · 99f375f · 99f375f
1 parent a722740
commit 99f375f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py
@@ -1827,8 +1827,8 @@ def test_groupby_apply_preserves_column_order(self):
     df = GROUPBY_DF
 
     self._run_test(
-        lambda df: df[['foo', 'group', 'bar']].groupby('group').apply(
-            lambda x: x),
+        lambda df: df[['foo', 'group', 'bar']].groupby(
+            'group', group_keys=False).apply(lambda x: x),
         df)
 
   def test_groupby_transform(self):