From 99f375f01e4e170b2be9909d727fe92d5e30044c Mon Sep 17 00:00:00 2001 From: Christopher Neffshade Date: Wed, 20 Sep 2023 18:49:31 +0000 Subject: [PATCH] Force old default for group_keys --- sdks/python/apache_beam/dataframe/frames_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py index b8b8cb733ef1..6b45ddf0533b 100644 --- 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):