AbstractVar+pyright should now work

patrick-kidger · Oct 3, 2023 · b0ff6c4 · b0ff6c4
1 parent 539a367
commit b0ff6c4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/equinox/_better_abstract.py b/equinox/_better_abstract.py
@@ -8,7 +8,6 @@
 import abc
 import dataclasses
 from typing import (
-    Annotated,
     ClassVar,
     Generic,
     get_args,
@@ -23,7 +22,9 @@
 
 
 if TYPE_CHECKING:
-    AbstractVar: TypeAlias = Annotated[_T, "AbstractVar"]
+    # Deliberately confuse pyright into treating this as `Unknown`.
+    # Then it won't complain when folks override with a concrete variable in a subclass.
+    AbstractVar: TypeAlias = getattr(abc, "foo" + "bar")  # pyright: ignore
     from typing import ClassVar as AbstractClassVar
 else: