diff options
Diffstat (limited to 'pylint/extensions/mccabe.py')
-rw-r--r-- | pylint/extensions/mccabe.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/pylint/extensions/mccabe.py b/pylint/extensions/mccabe.py index 27d246c1e..d4e752f48 100644 --- a/pylint/extensions/mccabe.py +++ b/pylint/extensions/mccabe.py @@ -13,6 +13,8 @@ """Module to add McCabe checker class for pylint. """ +from typing import TYPE_CHECKING + from astroid import nodes from mccabe import PathGraph as Mccabe_PathGraph from mccabe import PathGraphingAstVisitor as Mccabe_PathGraphingAstVisitor @@ -21,6 +23,9 @@ from pylint import checkers from pylint.checkers.utils import check_messages from pylint.interfaces import HIGH, IAstroidChecker +if TYPE_CHECKING: + from pylint.lint import PyLinter + class PathGraph(Mccabe_PathGraph): def __init__(self, node): @@ -194,10 +199,5 @@ class McCabeMethodChecker(checkers.BaseChecker): ) -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(McCabeMethodChecker(linter)) |