add pyarrow-stubs. add comment about check_array_indexer

Dr-Irv · Dr-Irv · commit 377872cb5022 · 2025-10-03T18:23:57.000-04:00
diff --git a/pandas-stubs/core/arrays/arrow/__init__.pyi b/pandas-stubs/core/arrays/arrow/__init__.pyi
diff --git a/pandas-stubs/core/arrays/arrow/dtype.pyi b/pandas-stubs/core/arrays/arrow/dtype.pyi
@@ -1,4 +1,4 @@
-import pyarrow as pa  # pyright: ignore[reportMissingTypeStubs]
+import pyarrow as pa
 
 from pandas._libs.missing import NAType
 
diff --git a/pandas-stubs/core/indexers/utils.pyi b/pandas-stubs/core/indexers/utils.pyi
@@ -6,6 +6,7 @@ from pandas._typing import (
     np_ndarray_int,
 )
 
+# Docs may not be right. See pandas-dev/pandas#62562
 @overload
 def check_array_indexer(array: AnyArrayLike, indexer: int) -> int: ...
 @overload
diff --git a/pyproject.toml b/pyproject.toml
@@ -68,6 +68,7 @@ types-python-dateutil = ">=2.8.19"
 beautifulsoup4 = ">=4.14.2"
 html5lib = ">=1.1"
 python-calamine = ">=0.2.0"
+pyarrow-stubs = { version = ">=20.0.0.20250928", python = "<4" }
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/tests/test_dtypes.py b/tests/test_dtypes.py
@@ -20,7 +20,7 @@
     BooleanArray,
     IntegerArray,
 )
-import pyarrow as pa  # pyright: ignore[reportMissingTypeStubs]
+import pyarrow as pa
 from typing_extensions import assert_type
 
 from pandas._typing import Scalar
diff --git a/tests/test_pandas.py b/tests/test_pandas.py
@@ -678,7 +678,7 @@ def test_crosstab() -> None:
 def test_arrow_dtype() -> None:
     pytest.importorskip("pyarrow")
 
-    import pyarrow as pa  # pyright: ignore[reportMissingTypeStubs]
+    import pyarrow as pa
 
     check(
         assert_type(

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-import pyarrow as pa # pyright: ignore[reportMissingTypeStubs]`
	`1`	`+import pyarrow as pa`
`2`	`2`
`3`	`3`	`from pandas._libs.missing import NAType`
`4`	`4`
Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@ from pandas._typing import (`
`6`	`6`	`np_ndarray_int,`
`7`	`7`	`)`
`8`	`8`
	`9`	`+# Docs may not be right. See pandas-dev/pandas#62562`
`9`	`10`	`@overload`
`10`	`11`	`def check_array_indexer(array: AnyArrayLike, indexer: int) -> int: ...`
`11`	`12`	`@overload`
Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,7 @@`
`20`	`20`	`BooleanArray,`
`21`	`21`	`IntegerArray,`
`22`	`22`	`)`
`23`		`-import pyarrow as pa # pyright: ignore[reportMissingTypeStubs]`
	`23`	`+import pyarrow as pa`
`24`	`24`	`from typing_extensions import assert_type`
`25`	`25`
`26`	`26`	`from pandas._typing import Scalar`