From b7775678a3b58446c8de8106863658864cb52d51 Mon Sep 17 00:00:00 2001
From: Marius van Niekerk <marius.v.niekerk@gmail.com>
Date: Sat, 19 Oct 2024 00:59:25 -0400
Subject: [PATCH] add list[str] overload to rename_columns (#119)
---
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: ...