NVIDIA / NVIDIA/cudf

[FEA] Support `pl.corr` in cudf-polars

Open
#20,564 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

```python
import polars as pl
pl.Config.set_verbose(True)

lf = pl.LazyFrame(
{
"a": [1, 2, 3, 4, 5, 6],
"b": [1, 2, 3, 4, 5, 6],
"g": ["x", "x", "x", "y", "y", "y"],
}
)

print(lf.select(pl.corr('a', 'b')).collect(engine='gpu'))
print(lf.group_by('g').agg(pl.corr('a', 'b')).collect(engine='gpu'))

"""
/opt/miniconda3_py312/lib/python3.12/site-packages/polars/lazyframe/frame.py:2407: PerformanceWarning: Query execution with GPU not possible: unsupported operations.
The errors were:
- NotImplementedError: corr
return wrap_df(ldf.collect(engine, callback))
"""
````

The cudf-polars version is '25.12.00a362'

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.