Оценить выражение внутри пользовательского класса в полярах ⇐ Python

[Anonymous]

Я пытаюсь расширить функциональность поляров для управления категориями Enum. Я следую этому руководству и этому разделу документации

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

orig_df = pl.DataFrame({
'idx': pl.int_range(5, eager=True),
'orig_series': pl.Series(['Alpha', 'Omega', 'Alpha', 'Beta', 'Gamma'],
dtype=pl.Enum(['Alpha', 'Beta', 'Gamma', 'Omega']))})

@pl.api.register_expr_namespace('fct')
class CustomEnumMethodsCollection:
def __init__(self, expr: pl.Expr):
self._expr = expr

def rev(self) -> pl.Expr:
cats = self._expr.cat.get_categories()
tmp_sr = self._expr.cast(pl.Categorical)
return tmp_sr.cast(dtype=pl.Enum(cats.str.reverse()))

(orig_df
.with_columns(rev_series=pl.col("orig_series").fct.rev())
)

Эта ошибка с TypeError: конструктор серии вызывается с неподдерживаемым типом «Expr» для значений

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

 parameter

потому что кошки — это невычисляемое выражение, а не список или серия, как этого ожидает pl.Enum(dtype=). Как мне оценить кошек в фактический список/серию, чтобы предоставить новые категории для моего метода cast(pl.Enum)?


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