NVIDIA / NVIDIA/cudf

[FEA] Support `filter` in groupby/rolling for cudf-polars

Open
#18,840 1 comment 0 reactions 1 assignee Claimed by @Matt711 View on GitHub
cudf-polars feature request Python
Dominant language
C++
Stars
9.8k
Forks
1.1k
Avg merge
3d 6m
Merged PRs (30d)
278

Description

```python
import numpy as np
import polars as pl

lf = pl.LazyFrame(
{
"a": [None, 2., 3.],
"b": [1, 1, 2],
}
)
with pl.Config() as cfg:
cfg.set_verbose(True)
print(lf.group_by('b').agg(pl.col('a').filter(pl.col('a')>2).sum()).collect(engine='gpu'))

"""
The errors were:
- NotImplementedError: No support for in groupby/rolling
return wrap_df(ldf.collect(engine, callback))
keys/aggregates are not partitionable: running default HASH AGGREGATION
shape: (2, 2)
┌─────┬─────┐
│ b ┆ a │
│ --- ┆ --- │
│ i64 ┆ f64 │
╞═════╪═════╡
│ 2 ┆ 3.0 │
│ 1 ┆ 0.0 │
└─────┴─────┘
"""
```

The cuDF version is '25.06.00a348'.

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.