🩹 python-version-dependent @override for __getstate__ (#317)

Author: jorenham

src/numpy-stubs/polynomial/_polybase.pyi

@@ -1,4 +1,5 @@
 import abc
+import sys
 from collections.abc import Iterator, Mapping, Sequence
 from typing import Any, ClassVar, Final, Literal, SupportsIndex, TypeAlias, overload
 from typing_extensions import Self, TypeIs, TypeVar, override
@@ -96,7 +97,14 @@ class ABCPolyBase(abc.ABC):
     def __rdivmod__(self, x: _AnyOther, /) -> tuple[Self, Self]: ...
     def __len__(self, /) -> int: ...
     def __iter__(self, /) -> Iterator[np.inexact | object]: ...
-    def __getstate__(self, /) -> dict[str, Any]: ...
+
+    if sys.version_info >= (3, 11):
+        @override
+        def __getstate__(self, /) -> dict[str, Any]: ...
+
+    else:
+        def __getstate__(self, /) -> dict[str, Any]: ...
+
     def __setstate__(self, dict: dict[str, Any], /) -> None: ...
 
     #

src/numpy-stubs/random/_generator.pyi

@@ -1,4 +1,5 @@
 import builtins
+import sys
 from collections.abc import Callable
 from typing import Any, Literal, TypeAlias, overload
 from typing_extensions import TypeVar, override
@@ -61,10 +62,19 @@ _ToRNG: TypeAlias = (
 
 class Generator:
     def __init__(self, /, bit_generator: BitGenerator[Any]) -> None: ...
-    def __getstate__(self) -> None: ...
+
+    if sys.version_info >= (3, 11):
+        @override
+        def __getstate__(self) -> None: ...
+
+    else:
+        def __getstate__(self) -> None: ...
+
     def __setstate__(self, state: dict[str, Any] | None) -> None: ...
     @override
     def __reduce__(self) -> tuple[Callable[[BitGenerator], Generator], tuple[BitGenerator], None]: ...
+
+    #
     @property
     def bit_generator(self) -> BitGenerator: ...
     def spawn(self, /, n_children: int) -> list[Generator]: ...

src/numpy-stubs/random/mtrand.pyi

@@ -1,4 +1,5 @@
 import builtins
+import sys
 from collections.abc import Callable
 from typing import Any, Final, Literal, TypeAlias, overload
 from typing_extensions import TypeVar, override
@@ -94,10 +95,19 @@ class RandomState:
     _bit_generator: Final[BitGenerator]
 
     def __init__(self, /, seed: _ArrayLikeInt_co | BitGenerator[Any] | None = ...) -> None: ...
-    def __getstate__(self) -> dict[str, Any]: ...
+
+    if sys.version_info >= (3, 11):
+        @override
+        def __getstate__(self) -> dict[str, Any]: ...
+
+    else:
+        def __getstate__(self) -> dict[str, Any]: ...
+
     def __setstate__(self, /, state: dict[str, Any]) -> None: ...
     @override
     def __reduce__(self) -> tuple[Callable[[BitGenerator], RandomState], tuple[BitGenerator], dict[str, Any]]: ...
+
+    #
     @overload
     def get_state(self, /, legacy: Literal[False]) -> dict[str, Any]: ...
     @overload