dask-sql
dask-sql copied to clipboard
[ENH] multi-argument grouping support
Is your feature request related to a problem? Please describe.
queries using grouping(x, y) cannot be run by dask-sql
for instance,
import dask_sql as ds
import pandas as pd
print(ds.__version__)
c = ds.Context()
c.create_table('tbl', pd.DataFrame({'x': [1, 1, 2, 2, 2], 'y': [2, 2, 1, 2, 2], 'z': [1, 3, 10, 3, 11]}))
c.sql("""
select x, y, sum(z), grouping(x, y)
from tbl
group by x, y
""")
results in,
2022.10.1
...
ParsingException: Plan("The function Grouping expects 1 arguments, but 2 were provided")