summaryrefslogtreecommitdiff
path: root/pylint/extensions/consider_ternary_expression.py
diff options
context:
space:
mode:
Diffstat (limited to 'pylint/extensions/consider_ternary_expression.py')
-rw-r--r--pylint/extensions/consider_ternary_expression.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/pylint/extensions/consider_ternary_expression.py b/pylint/extensions/consider_ternary_expression.py
index 3e3e87c6b..6dabe3613 100644
--- a/pylint/extensions/consider_ternary_expression.py
+++ b/pylint/extensions/consider_ternary_expression.py
@@ -1,10 +1,15 @@
"""Check for if / assign blocks that can be rewritten with if-expressions."""
+from typing import TYPE_CHECKING
+
from astroid import nodes
from pylint.checkers import BaseChecker
from pylint.interfaces import IAstroidChecker
+if TYPE_CHECKING:
+ from pylint.lint import PyLinter
+
class ConsiderTernaryExpressionChecker(BaseChecker):
@@ -44,10 +49,5 @@ class ConsiderTernaryExpressionChecker(BaseChecker):
self.add_message("consider-ternary-expression", node=node)
-def register(linter):
- """Required method to auto register this checker.
-
- :param linter: Main interface object for Pylint plugins
- :type linter: Pylint object
- """
+def register(linter: "PyLinter") -> None:
linter.register_checker(ConsiderTernaryExpressionChecker(linter))