diff options
Diffstat (limited to 'pylint/extensions/set_membership.py')
-rw-r--r-- | pylint/extensions/set_membership.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/pylint/extensions/set_membership.py b/pylint/extensions/set_membership.py index c63e268b3..6ba5166c5 100644 --- a/pylint/extensions/set_membership.py +++ b/pylint/extensions/set_membership.py @@ -1,9 +1,13 @@ +from typing import TYPE_CHECKING + from astroid import nodes from pylint.checkers import BaseChecker from pylint.checkers.utils import check_messages from pylint.interfaces import IAstroidChecker -from pylint.lint import PyLinter + +if TYPE_CHECKING: + from pylint.lint import PyLinter class SetMembershipChecker(BaseChecker): @@ -21,7 +25,7 @@ class SetMembershipChecker(BaseChecker): ), } - def __init__(self, linter: PyLinter) -> None: + def __init__(self, linter: "PyLinter") -> None: """Initialize checker instance.""" super().__init__(linter=linter) @@ -43,5 +47,5 @@ class SetMembershipChecker(BaseChecker): self.add_message("use-set-for-membership", node=comparator) -def register(linter: PyLinter) -> None: +def register(linter: "PyLinter") -> None: linter.register_checker(SetMembershipChecker(linter)) |