diff options
Diffstat (limited to 'pylint/extensions/consider_ternary_expression.py')
-rw-r--r-- | pylint/extensions/consider_ternary_expression.py | 12 |
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)) |