From c70177752f309311d7e019345e26c4d2e2732ec6 Mon Sep 17 00:00:00 2001 From: Marius van Niekerk Date: Fri, 18 Oct 2024 11:53:54 -0400 Subject: [PATCH] add list[str] overload to rename_columns see: https://arrow.apache.org/docs/python/generated/pyarrow.RecordBatch.html\#pyarrow.RecordBatch.rename_columns --- pyarrow-stubs/__lib_pxi/table.pyi | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/pyarrow-stubs/__lib_pxi/table.pyi b/pyarrow-stubs/__lib_pxi/table.pyi index 80c875f..9c1fb16 100644 --- a/pyarrow-stubs/__lib_pxi/table.pyi +++ b/pyarrow-stubs/__lib_pxi/table.pyi @@ -453,6 +453,9 @@ class RecordBatch(_Tabular[Array]): def get_total_buffer_size(self) -> int: ... def __sizeof__(self) -> int: ... def set_column(self, i: int, field_: str | Field, column: Array | list) -> Self: ... + @overload + def rename_columns(self, names: list[str]) -> Self: ... + @overload def rename_columns(self, names: dict[str, str]) -> Self: ... def serialize(self, memory_pool: MemoryPool | None = None) -> Buffer: ... def slice(self, offset: int = 0, length: int | None = None) -> Self: ... @@ -566,6 +569,9 @@ class Table(_Tabular[ChunkedArray]): def get_total_buffer_size(self) -> int: ... def __sizeof__(self) -> int: ... def set_column(self, i: int, field_: str | Field, column: Array | list) -> Self: ... + @overload + def rename_columns(self, names: list[str]) -> Self: ... + @overload def rename_columns(self, names: dict[str, str]) -> Self: ... def drop(self, columns: str | list[str]) -> Self: ... def group_by(self, keys: str | list[str], use_threads: bool = True) -> TableGroupBy: ...