Агрегация Polars group_by по сумме возвращает ошибку при использовании словаря ⇐ Python

[Anonymous]

Я пытаюсь выполнить агрегацию из полярного DataFrame. Но я не получаю того, что ожидаю.
Это минимальное повторение проблемы:

Код: Выделить всё

import polars as pl

# Create a DataFrame
df = pl.DataFrame({"category": ["A", "A", "B", "B", "B"],
"value": [1., 2., 3., 4., 5.]})

# Group by 'category' and sum 'value'
result = df.group_by("category").agg({"value": pl.sum})

# Print the result
print(result)

И я получаю:

Код: Выделить всё

TypeError: specifying aggregations as a dictionary is not supported

но я бы хотел получить:

Код: Выделить всё

┌──────────┬─────────────────┐
│ category ┆ value           │
│ ---      ┆ ---             │
│ str      ┆ list[f64]       │
╞══════════╪═════════════════╡
│ A        ┆ 3.0             │
│ B        ┆ 12.0            │
└──────────┴─────────────────┘

Есть идеи, в чем проблема? заранее спасибо.

Подробнее здесь: https://stackoverflow.com/questions/760 ... dictionary