diff --git a/python/dask_cudf/dask_cudf/groupby.py b/python/dask_cudf/dask_cudf/groupby.py index 336fdaf009c..2ec457018d9 100644 --- a/python/dask_cudf/dask_cudf/groupby.py +++ b/python/dask_cudf/dask_cudf/groupby.py @@ -143,7 +143,7 @@ def aggregate(self, arg, split_every=None, split_out=1): sep=self.sep, sort=self.sort, as_index=self.as_index, - ) + )[self._slice] return super().aggregate( arg, split_every=split_every, split_out=split_out diff --git a/python/dask_cudf/dask_cudf/tests/test_groupby.py b/python/dask_cudf/dask_cudf/tests/test_groupby.py index 356567fdef0..5cb681fff55 100644 --- a/python/dask_cudf/dask_cudf/tests/test_groupby.py +++ b/python/dask_cudf/dask_cudf/tests/test_groupby.py @@ -48,10 +48,7 @@ def test_groupby_basic_aggs(aggregation): "func", [ lambda df: df.groupby("x").agg({"y": "max"}), - pytest.param( - lambda df: df.groupby("x").y.agg(["sum", "max"]), - marks=pytest.mark.skip, - ), + lambda df: df.groupby("x").y.agg(["sum", "max"]), ], ) def test_groupby_agg(func): @@ -98,7 +95,6 @@ def test_groupby_agg_empty_partition(tmpdir, split_out): dd.assert_eq(gb.compute().sort_index(), expect) -@pytest.mark.xfail(reason="cudf issues") @pytest.mark.parametrize( "func", [lambda df: df.groupby("x").std(), lambda df: df.groupby("x").y.std()], @@ -115,13 +111,9 @@ def test_groupby_std(func): ddf = dask_cudf.from_cudf(gdf, npartitions=5) - a = func(gdf.to_pandas()) + a = func(gdf).to_pandas() b = func(ddf).compute().to_pandas() - a.index.name = None - a.name = None - b.index.name = None - dd.assert_eq(a, b) @@ -129,9 +121,7 @@ def test_groupby_std(func): "func", [ lambda df: df.groupby("x").agg({"y": "collect"}), - pytest.param( - lambda df: df.groupby("x").y.agg("collect"), marks=pytest.mark.skip - ), + lambda df: df.groupby("x").y.agg("collect"), ], ) def test_groupby_collect(func):