type override better

.github/workflows/ci.yml

@@ -27,5 +27,6 @@ jobs:
         run: pytest tests
       - name: Run mypy
         run:  mypy overrides
-      - name: Run mypy static tests
+      - name: Run mypy static tests (Python=3.10)
+        if: matrix.python == '3.10'
         run: ./check_mypy.sh

mypy_passes/passing_rewrite.py

@@ -0,0 +1,15 @@
+""""Bug."""
+from typing import Callable, TypeVar, ParamSpec
+
+P = ParamSpec("P")
+T = TypeVar("T")
+
+try:
+    from overrides import override
+except ModuleNotFoundError:
+
+    def override(func: Callable[P, T]) -> Callable[P, T]:  # type: ignore[no-redef]
+        return func
+
+
+print(override)

overrides/__init__.py

@@ -5,9 +5,8 @@
     from overrides.final import final
 else:
     from typing import final
-from overrides.overrides import __VERSION__, overrides
+from overrides.overrides import __VERSION__, overrides, override
 
-override = overrides
 
 __all__ = [
     "__VERSION__",

overrides/overrides.py

@@ -94,6 +94,37 @@ def method(self):
         )
 
 
+@overload
+def override(
+    method: None = None,
+    *,
+    check_signature: bool = True,
+    check_at_runtime: bool = False,
+) -> _DecoratorMethod:
+    ...
+
+
+@overload
+def override(
+    method: _WrappedMethod,
+    *,
+    check_signature: bool = True,
+    check_at_runtime: bool = False,
+) -> _WrappedMethod:
+    ...
+
+
+def override(
+    method: Optional[_WrappedMethod] = None,
+    *,
+    check_signature: bool = True,
+    check_at_runtime: bool = False,
+) -> Union[_DecoratorMethod, _WrappedMethod]:
+    return overrides(
+        method, check_signature=check_signature, check_at_runtime=check_at_runtime
+    )
+
+
 def _overrides(
     method: _WrappedMethod,
     check_signature: bool,