diff options
Diffstat (limited to 'pylint/extensions/comparetozero.py')
-rw-r--r-- | pylint/extensions/comparetozero.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/pylint/extensions/comparetozero.py b/pylint/extensions/comparetozero.py index e543f1ccf..592e15a5b 100644 --- a/pylint/extensions/comparetozero.py +++ b/pylint/extensions/comparetozero.py @@ -13,7 +13,7 @@ """Looks for comparisons to zero.""" import itertools -from typing import Any, Iterable +from typing import TYPE_CHECKING, Any, Iterable import astroid from astroid import nodes @@ -21,6 +21,9 @@ from astroid import nodes from pylint import checkers, interfaces from pylint.checkers import utils +if TYPE_CHECKING: + from pylint.lint import PyLinter + def _is_constant_zero(node): return isinstance(node, astroid.Const) and node.value == 0 @@ -77,6 +80,5 @@ class CompareToZeroChecker(checkers.BaseChecker): self.add_message("compare-to-zero", node=node) -def register(linter): - """Required method to auto register this checker.""" +def register(linter: "PyLinter") -> None: linter.register_checker(CompareToZeroChecker(linter)) |