Исключение паники при расчете совокупной оценки для каждой группы в полярах ⇐ Python

[Anonymous]

Обновление. Это было исправлено с помощью pull/6435, и теперь запрос выполняется должным образом.

Я хочу вычислять совокупные значения (например, среднее значение, количество уникальных значений и режим) для каждой группы, а в некоторых группах есть только значение «Нет».
Когда я вычисляю значения для столбцов, в которых нет значения «Нет», я получаю правильный результат. Однако, когда я указываю столбцы, в некоторых группах которых есть только значение «Нет», я получаю следующую ошибку.
Как я могу вычислить значения?
Например,
df = pl.DataFrame({'key': [1, 1, 2, 1, 2, 3, 3, 3, 2],
'col1': [1, 2, 1, 3, 1, 2, 3, 4, 1],
'col2': [1, 2, None, 3, None, None, 3, 4, None]})

df.group_by('key').agg(
pl.col('col2').cumulative_eval(pl.element().mean()).name.suffix('_cummean')
).explode('col2_cummean')

PanicException: called `Result::unwrap()` on an `Err` value: SchemaMisMatch(Borrowed("cannot unpack Series; data types don't match"))

Ожидаемый результат:
shape: (9, 2)
┌─────┬──────────────┐
│ key ┆ col2_cummean │
│ --- ┆ --- │
│ i64 ┆ f64 │
╞═════╪══════════════╡
│ 1 ┆ 1.0 │
│ 1 ┆ 1.5 │
│ 1 ┆ 2.0 │
│ 2 ┆ null │
│ 2 ┆ null │
│ 2 ┆ null │
│ 3 ┆ null │
│ 3 ┆ 3.0 │
│ 3 ┆ 3.5 │
└─────┴──────────────┘


Подробнее здесь: https://stackoverflow.com/questions/752 ... -in-polars