fix: error comes from pyright

Author: cmp0xff

pandas-stubs/core/generic.pyi

@@ -430,6 +430,6 @@ class NDFrame(indexing.IndexingMixin):
         origin: TimeGrouperOrigin | TimestampConvertibleTypes = "start_day",
         offset: TimedeltaConvertibleTypes | None = None,
         group_keys: _bool = False,
-    ) -> DatetimeIndexResampler[Self]: ...
+    ) -> DatetimeIndexResampler[Self]: ...  # pyrefly: ignore[bad-specialization]
     @final
     def take(self, indices: TakeIndexer, axis: Axis = 0, **kwargs: Any) -> Self: ...

pandas-stubs/core/groupby/generic.pyi

@@ -229,7 +229,7 @@ class DFCallable3(Protocol[P]):  # ty: ignore[invalid-argument-type]
 class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
     # error: Overload 3 for "apply" will never be used because its parameters overlap overload 1
     @overload  # type: ignore[override]
-    def apply(
+    def apply(  # pyrefly: ignore[bad-override]
         self,
         func: DFCallable1[P],
         /,

pandas-stubs/core/groupby/groupby.pyi

@@ -255,7 +255,9 @@ class GroupBy(BaseGroupBy[NDFrameT]):
     def bfill(self, limit: int | None = ...) -> NDFrameT: ...
     @final
     @property
-    def nth(self) -> GroupByNthSelector[Self]: ...
+    def nth(
+        self,
+    ) -> GroupByNthSelector[Self]: ...  # pyrefly: ignore[bad-specialization]
     @final
     def quantile(
         self,

pandas-stubs/core/indexes/base.pyi

@@ -279,7 +279,7 @@ class Index(IndexOpsMixin[S1]):
     @property
     def str(
         self,
-    ) -> StringMethods[
+    ) -> StringMethods[  # pyrefly: ignore[bad-specialization]
         Self,
         MultiIndex,
         np_1darray[np.bool],

pandas-stubs/core/series.pyi

@@ -1198,7 +1198,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     @property
     def str(
         self,
-    ) -> StringMethods[
+    ) -> StringMethods[  # pyrefly: ignore[bad-specialization]
         Self,
         DataFrame,
         Series[bool],

pyproject.toml

@@ -41,15 +41,15 @@ pandas = "2.3.2"
 pyarrow = ">=10.0.1"
 pytest = ">=7.1.2"
 pyright = ">=1.1.405"
-ty = "^0.0.1a9"
-pyrefly = "^0.21.0"
+ty = ">=0.0.1a20"
+pyrefly = ">=0.32.0"
 poethepoet = ">=0.16.5"
 loguru = ">=0.6.0"
 typing-extensions = ">=4.4.0"
 matplotlib = ">=3.10.1"
 pre-commit = ">=2.19.0"
-black = ">=23.3.0"
-isort = ">=5.12.0"
+black = ">=25.1.0"
+isort = ">=6.0.1"
 openpyxl = ">=3.0.10"
 tables = { version = ">=3.10.1", python = "<4" }
 lxml = ">=4.9.1"

tests/indexes/arithmetic/test_add.py

@@ -1,5 +1,6 @@
+from typing import Any
+
 import numpy as np
-from numpy import typing as npt  # noqa: F401
 import pandas as pd
 from typing_extensions import (
     Never,
@@ -61,18 +62,22 @@ def test_add_i_numpy_array() -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__radd__` cannot override. At runtime, they return
     # `Index`s.
-    # `mypy` thinks the return types are `Any`, which is a bug.
+    # microsoft/pyright#10924
     check(
-        assert_type(b + left_i, "npt.NDArray[np.bool_]"), pd.Index  # type: ignore[assert-type]
+        assert_type(b + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
     check(
-        assert_type(i + left_i, "npt.NDArray[np.int64]"), pd.Index  # type: ignore[assert-type]
+        assert_type(i + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
     check(
-        assert_type(f + left_i, "npt.NDArray[np.float64]"), pd.Index  # type: ignore[assert-type]
+        assert_type(f + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
     check(
-        assert_type(c + left_i, "npt.NDArray[np.complex128]"), pd.Index  # type: ignore[assert-type]
+        assert_type(c + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
 
 

tests/indexes/arithmetic/test_sub.py

@@ -1,7 +1,6 @@
-from typing import NoReturn
+from typing import Any
 
 import numpy as np
-from numpy import typing as npt  # noqa: F401
 import pandas as pd
 from typing_extensions import assert_type
 
@@ -59,16 +58,22 @@ def test_sub_i_numpy_array() -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rsub__` cannot override. At runtime, they return
     # `Index`s.
-    # `mypy` thinks the return types are `Any`, which is a bug.
-    check(assert_type(b - left_i, NoReturn), pd.Index)  # type: ignore[assert-type]
+    # microsoft/pyright#10924
     check(
-        assert_type(i - left_i, "npt.NDArray[np.int64]"), pd.Index  # type: ignore[assert-type]
+        assert_type(b - left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
     check(
-        assert_type(f - left_i, "npt.NDArray[np.float64]"), pd.Index  # type: ignore[assert-type]
+        assert_type(i - left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
     check(
-        assert_type(c - left_i, "npt.NDArray[np.complex128]"), pd.Index  # type: ignore[assert-type]
+        assert_type(f - left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
+    )
+    check(
+        assert_type(c - left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
     )
 
 

tests/series/arithmetic/test_add.py

@@ -4,7 +4,6 @@
 )
 
 import numpy as np
-from numpy import typing as npt  # noqa: F401
 import pandas as pd
 from typing_extensions import (
     Never,
@@ -95,18 +94,22 @@ def test_add_i_numpy_array() -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__radd__` cannot override. At runtime, they return
     # `Series`s.
-    # `mypy` thinks the return types are `Any`, which is a bug.
+    # microsoft/pyright#10924
     check(
-        assert_type(b + left_i, "npt.NDArray[np.bool_]"), pd.Series  # type: ignore[assert-type]
+        assert_type(b + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
     check(
-        assert_type(i + left_i, "npt.NDArray[np.int64]"), pd.Series  # type: ignore[assert-type]
+        assert_type(i + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
     check(
-        assert_type(f + left_i, "npt.NDArray[np.float64]"), pd.Series  # type: ignore[assert-type]
+        assert_type(f + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
     check(
-        assert_type(c + left_i, "npt.NDArray[np.complex128]"), pd.Series  # type: ignore[assert-type]
+        assert_type(c + left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
 
     check(assert_type(left_i.add(b), pd.Series), pd.Series)

tests/series/arithmetic/test_mul.py

@@ -1,5 +1,6 @@
+from typing import Any
+
 import numpy as np
-from numpy import typing as npt  # noqa: F401
 import pandas as pd
 from typing_extensions import assert_type
 
@@ -76,18 +77,22 @@ def test_mul_numpy_array() -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rmul__` cannot override. At runtime, they return
     # `Series`s.
-    # `mypy` thinks the return types are `Any`, which is a bug.
+    # microsoft/pyright#10924
     check(
-        assert_type(b * left_i, "npt.NDArray[np.bool_]"), pd.Series  # type: ignore[assert-type]
+        assert_type(b * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
     check(
-        assert_type(i * left_i, "npt.NDArray[np.int64]"), pd.Series  # type: ignore[assert-type]
+        assert_type(i * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
     check(
-        assert_type(f * left_i, "npt.NDArray[np.float64]"), pd.Series  # type: ignore[assert-type]
+        assert_type(f * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
     check(
-        assert_type(c * left_i, "npt.NDArray[np.complex128]"), pd.Series  # type: ignore[assert-type]
+        assert_type(c * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Series,
     )
 
     check(assert_type(left_i.mul(b), pd.Series), pd.Series)