NVIDIA / NVIDIA/cudf

[FEA] Pass `keys_are_sorted` in intermediate groupby operation in cudf_polars

Open
#21,911 0 comments 0 reactions 0 assignees View on GitHub
cudf-polars feature request
Dominant language
C++
Stars
9.8k
Forks
1.1k
Avg merge
3d 6m
Merged PRs (30d)
278

Description

(Identified by Claude)

Similar to what was already done in regular groupby operations. If the input column is sorted, `keys_are_sorted=plc.types.Sorted.YES` should be passed into intermediate groupby operations like:

https://github.com/rapidsai/cudf/blob/01c7adae10d12da7a175fb616300f8190522b6e0/python/cudf_polars/cudf_polars/dsl/expressions/unary.py#L283-L284

And if applicable like what was done in https://github.com/rapidsai/cudf/pull/21871, mark resulting resulting columns of the groupby keys as sorted like in

https://github.com/rapidsai/cudf/blob/01c7adae10d12da7a175fb616300f8190522b6e0/python/cudf_polars/cudf_polars/dsl/expressions/rolling.py#L744

https://github.com/rapidsai/cudf/blob/01c7adae10d12da7a175fb616300f8190522b6e0/python/cudf_polars/cudf_polars/dsl/expressions/rolling.py#L291

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.