summaryrefslogtreecommitdiff
path: root/pylint/extensions/set_membership.py
diff options
context:
space:
mode:
Diffstat (limited to 'pylint/extensions/set_membership.py')
-rw-r--r--pylint/extensions/set_membership.py10
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))