diff --git a/pyarrow-stubs/__lib_pxi/table.pyi b/pyarrow-stubs/__lib_pxi/table.pyi index 733733a..8a84a7f 100644 --- a/pyarrow-stubs/__lib_pxi/table.pyi +++ b/pyarrow-stubs/__lib_pxi/table.pyi @@ -52,7 +52,7 @@ from .io import Buffer from .ipc import RecordBatchReader from .scalar import Int64Scalar, Scalar from .tensor import Tensor -from .types import DataType, _AsPyType, _BasicDataType, _DataTypeT +from .types import _AsPyType, _BasicDataType, _DataType_CoT, _DataTypeT _Scalar_CoT = TypeVar("_Scalar_CoT", bound=Scalar, covariant=True) @@ -60,7 +60,7 @@ class ChunkedArray(_PandasConvertible[pd.Series], Generic[_Scalar_CoT]): @property def data(self) -> Self: ... @property - def type(self) -> DataType: ... + def type(self: ChunkedArray[Scalar[_DataType_CoT]]) -> _DataType_CoT: ... def length(self) -> int: ... __len__ = length def to_string(