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: ...