Docstring для Polars API -регистрации пространство имен ⇐ Python

[Anonymous]

Пример Python Polars API udf Регистрация имени: < /p>

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

import polars as pl
from pydantic import ValidationError, validate_call

@pl.api.register_dataframe_namespace("test")
class TestOps:
def __init__(self, df: pl.DataFrame) -> None:
self._df = df

@validate_call
def add_lit_col(self, lit_int: int = 5) -> pl.DataFrame:
""" Add a literal integer defined by `lit_int` to the dataframe
"""
return self._df.with_columns(pl.lit(lit_int))

pl.DataFrame(
data=["aaa", "bbb", "ccc", "ddd", "eee", "fff"],
).test.add_lit_col("foo")

Pydantic валидация работает, и когда я распечатаю строку DOC для ADD_LIT_COL Метод. Он также правильно показывает:

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

TestOps.add_lit_col.__doc__

out:
' Add a literal integer defined by `lit_int` to the dataframe\n        '

Но когда я падаю над как зарегистрированное пространство имен для pl.dataframe тест модуля , а также метод add_lit_col , он не показывает Строка и параметры DOC: (function) add_lit_col: любой


Подробнее здесь: https://stackoverflow.com/questions/794 ... -namespace