@@ -28,6 +28,7 @@ from typing import (
28
28
import numpy as np
29
29
import pandas as pd
30
30
31
+ from numpy .typing import NDArray
31
32
from pyarrow ._compute import (
32
33
CastOptions ,
33
34
CountOptions ,
@@ -525,7 +526,7 @@ class RecordBatch(_Tabular[Array]):
525
526
def serialize (self , memory_pool : MemoryPool | None = None ) -> Buffer : ...
526
527
def slice (self , offset : int = 0 , length : int | None = None ) -> Self : ...
527
528
def equals (self , other : Self , check_metadata : bool = False ) -> bool : ...
528
- def select (self , columns : list [str ] | Indices ) -> Self : ...
529
+ def select (self , columns : Iterable [str ] | Iterable [ int ] | NDArray [ np . str_ ] ) -> Self : ...
529
530
def cast (
530
531
self , target_schema : Schema , safe : bool | None = None , options : CastOptions | None = None
531
532
) -> Self : ...
@@ -592,7 +593,7 @@ JoinType: TypeAlias = Literal[
592
593
class Table (_Tabular [ChunkedArray [Any ]]):
593
594
def validate (self , * , full = False ) -> None : ...
594
595
def slice (self , offset = 0 , length = None ) -> Self : ...
595
- def select (self , columns : list [str ] | Indices ) -> Self : ...
596
+ def select (self , columns : Iterable [str ] | Iterable [ int ] | NDArray [ np . str_ ] ) -> Self : ...
596
597
def replace_schema_metadata (self , metadata : dict | None = None ) -> Self : ...
597
598
def flatten (self , memory_pool : MemoryPool | None = None ) -> Self : ...
598
599
def combine_chunks (self , memory_pool : MemoryPool | None = None ) -> Self : ...
0 commit comments