pyright @ py310

Author: cmp0xff

pandas-stubs/__init__.pyi

@@ -1,3 +1,5 @@
+import sys
+
 from pandas import (
     api as api,
     arrays as arrays,
@@ -47,7 +49,6 @@ from pandas.core.api import (
     UInt16Dtype as UInt16Dtype,
     UInt32Dtype as UInt32Dtype,
     UInt64Dtype as UInt64Dtype,
-    array as array,
     bdate_range as bdate_range,
     date_range as date_range,
     factorize as factorize,
@@ -157,4 +158,11 @@ from pandas.io.json._normalize import (
 from pandas.tseries import offsets as offsets
 from pandas.tseries.api import infer_freq as infer_freq
 
+if sys.version_info >= (3, 11):
+    from pandas.core.construction import array as array
+else:
+    from pandas.core.construction import (
+        array as array,  # pyright: ignore[reportUnknownVariableType]
+    )
+
 __version__: str

pandas-stubs/core/api.pyi

@@ -1,3 +1,5 @@
+import sys
+
 from pandas.core.algorithms import (
     factorize as factorize,
     unique as unique,
@@ -23,7 +25,6 @@ from pandas.core.arrays.integer import (
     UInt64Dtype as UInt64Dtype,
 )
 from pandas.core.arrays.string_ import StringDtype as StringDtype
-from pandas.core.construction import array as array
 from pandas.core.frame import DataFrame as DataFrame
 from pandas.core.groupby import (
     Grouper as Grouper,
@@ -80,3 +81,10 @@ from pandas.core.dtypes.missing import (
 
 from pandas.io.formats.format import set_eng_float_format as set_eng_float_format
 from pandas.tseries.offsets import DateOffset as DateOffset
+
+if sys.version_info >= (3, 11):
+    from pandas.core.construction import array as array
+else:
+    from pandas.core.construction import (
+        array as array,  # pyright: ignore[reportUnknownVariableType]
+    )

tests/arrays/test_arrays.py

@@ -1,11 +1,19 @@
+import sys
+
 from pandas.core.arrays.integer import IntegerArray
-from pandas.core.construction import array
 from typing_extensions import assert_type
 
 from pandas._libs.missing import NA
 
 from tests import check
 
+if sys.version_info >= (3, 11):
+    from pandas.core.construction import array
+else:
+    from pandas.core.construction import (
+        array,  # pyright: ignore[reportUnknownVariableType]
+    )
+
 
 def test_construction() -> None:
     check(assert_type(array([1]), IntegerArray), IntegerArray)

tests/arrays/test_cumul.py

@@ -1,11 +1,19 @@
+import sys
+
 from pandas.core.arrays.integer import IntegerArray
-from pandas.core.construction import array
 from typing_extensions import assert_type
 
 from pandas._libs.missing import NA
 
 from tests import check
 
+if sys.version_info >= (3, 11):
+    from pandas.core.construction import array
+else:
+    from pandas.core.construction import (
+        array,  # pyright: ignore[reportUnknownVariableType]
+    )
+
 
 def test_cumul_int64dtype() -> None:
     arr = array([1, NA, 2])

tests/arrays/test_extension_array.py

@@ -1,12 +1,13 @@
 # Test common ExtensionArray methods
+import sys
+
 import numpy as np
 import pandas as pd
 from pandas.core.arrays import ExtensionArray
 from pandas.core.arrays.integer import (
     Int32Dtype,
     IntegerArray,
 )
-from pandas.core.construction import array
 from typing_extensions import assert_type
 
 from pandas._typing import ArrayLike
@@ -17,6 +18,13 @@
     np_1darray_intp,
 )
 
+if sys.version_info >= (3, 11):
+    from pandas.core.construction import array
+else:
+    from pandas.core.construction import (
+        array,  # pyright: ignore[reportUnknownVariableType]
+    )
+
 
 def test_ea_common() -> None:
     # Note: `ExtensionArray` is abstract, so we use `IntegerArray` for the tests.