Avoid DeprecationWarning by setting include_groups=False in apply

pandas-dev · May 8, 2024 · 321b8ff · 321b8ff
1 parent c0c41f4
commit 321b8ff
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/pandas/tests/groupby/test_grouping.py b/pandas/tests/groupby/test_grouping.py
@@ -845,12 +845,9 @@ def test_groupby_tuple_keys_handle_multiindex(self):
             }
         )
         expected = df.sort_values(by=["category_tuple", "num1"])
-
-        msg = "DataFrameGroupBy.apply operated on the grouping columns"
-        with tm.assert_produces_warning(DeprecationWarning, match=msg):
-            result = df.groupby("category_tuple").apply(
-                lambda x: x.sort_values(by="num1")
-            )
+        result = df.groupby("category_tuple").apply(
+            lambda x: x.sort_values(by="num1"), include_groups=False
+        )
         expected = expected[result.columns]
         tm.assert_frame_equal(result.reset_index(drop=True), expected)