Skip to content

[FEA] Support grouped list aggregation and explode in cuDF-Polars #23595

Description

@rjzamora

Grouped list aggregation is not currently supported by the GPU engine, which also blocks workflows that aggregate values into lists and then explode them back into rows.

def test_grouped_list_aggregation_explode(engine: GPUEngine) -> None:
    lf = pl.LazyFrame(
        {
            "g": [1, 1, 2, 2, 2],
            "x": [10, 20, 30, 40, 50],
        }
    )

    q = lf.group_by("g").agg(pl.col("x")).explode("x")

    assert_gpu_result_equal(q, engine=engine, check_row_order=False)

Related broad inventory issue: #23151

Metadata

Metadata

Assignees

No one assigned

    Labels

    cudf-polarsIssues specific to cudf-polarsfeature requestNew feature or request

    Type

    No type

    Projects

    Status
    Todo

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions