From a6a8686ce2a5fa0ee9f112db28c684e6e8d9edba Mon Sep 17 00:00:00 2001 From: rjzamora Date: Thu, 23 May 2024 09:31:22 -0700 Subject: [PATCH] update test_groupby_grouper_dispatch --- dask/dataframe/tests/test_groupby.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/dask/dataframe/tests/test_groupby.py b/dask/dataframe/tests/test_groupby.py index c954ef335c8..ad16b7abc75 100644 --- a/dask/dataframe/tests/test_groupby.py +++ b/dask/dataframe/tests/test_groupby.py @@ -2991,9 +2991,8 @@ def test_groupby_grouper_dispatch(key): pd_grouper = grouper_dispatch(pdf)(key=key) gd_grouper = grouper_dispatch(gdf)(key=key) - # cuDF's numeric behavior aligns with numeric_only=True expect = pdf.groupby(pd_grouper).sum(numeric_only=True) - got = gdf.groupby(gd_grouper).sum() + got = gdf.groupby(gd_grouper).sum(numeric_only=True) assert_eq(expect, got)