add list[str] overload to rename_columns (#119)

zen-xu · Oct 19, 2024 · b777567 · b777567
1 parent 34a956f
commit b777567
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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: ...