Аргумент Polars Map_batches return_dtype для массивов ⇐ Python

[Anonymous]

Я применяю пользовательскую функцию (UDF) к кадру данных Polars с помощью функции map_batches (см. https://docs.pola.rs/user-guide/express ... umn-values).
Пользовательская функция возвращает массив типа int8.
Похоже, что из поляры v1.32 в аргументе return_dtype требуется функция map_batches (c.p. https://docs.pola.rs/api/python/stable/ ... tches.html)
Я пытаюсь правильно определить аргумент return_dtype. Я пробовал такие вещи, как return_dtype=pl.Array[pl.Int8]. Однако это не работает.
РЕДАКТИРОВАТЬ:
До появления Polars v1.32 схема моего фрейма данных была следующей (здесь показан только соответствующий столбец):

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

Schema([(...),
('signals', Array(Int8, shape=(9,)))])

Столбец сигналов будет получен из UDF, который возвращает массив numpy. UDF будет применен к группам кадра данных (т. е. используется операция над).

Подробнее здесь: https://stackoverflow.com/questions/797 ... for-arrays