diff --git a/mypy/traverser.py b/mypy/traverser.py index 4ce8332fed86..c4834c9acb6b 100644 --- a/mypy/traverser.py +++ b/mypy/traverser.py @@ -1,6 +1,7 @@ """Generic node traverser visitor""" from typing import List +from mypy_extensions import mypyc_attr from mypy.visitor import NodeVisitor from mypy.nodes import ( @@ -16,6 +17,7 @@ ) +@mypyc_attr(allow_interpreted_subclasses=True) class TraverserVisitor(NodeVisitor[None]): """A parse tree visitor that traverses the parse tree during visiting. diff --git a/mypy/visitor.py b/mypy/visitor.py index 09a6cea9106a..b98ec773bbe3 100644 --- a/mypy/visitor.py +++ b/mypy/visitor.py @@ -3,7 +3,7 @@ from abc import abstractmethod from typing import TypeVar, Generic from typing_extensions import TYPE_CHECKING -from mypy_extensions import trait +from mypy_extensions import trait, mypyc_attr if TYPE_CHECKING: # break import cycle only needed for mypy @@ -14,6 +14,7 @@ @trait +@mypyc_attr(allow_interpreted_subclasses=True) class ExpressionVisitor(Generic[T]): @abstractmethod def visit_int_expr(self, o: 'mypy.nodes.IntExpr') -> T: @@ -193,6 +194,7 @@ def visit_temp_node(self, o: 'mypy.nodes.TempNode') -> T: @trait +@mypyc_attr(allow_interpreted_subclasses=True) class StatementVisitor(Generic[T]): # Definitions @@ -310,6 +312,7 @@ def visit_exec_stmt(self, o: 'mypy.nodes.ExecStmt') -> T: @trait +@mypyc_attr(allow_interpreted_subclasses=True) class NodeVisitor(Generic[T], ExpressionVisitor[T], StatementVisitor[T]): """Empty base class for parse tree node visitors.