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