Fix test after merge with main

pylint-dev · Sep 3, 2021 · e4ada21 · e4ada21
1 parent 85fe213
commit e4ada21
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/extensions/test_check_docs.py b/tests/extensions/test_check_docs.py
@@ -26,7 +26,7 @@
 
 import astroid
 import pytest
-from astroid.nodes import ClassDef
+from astroid import nodes
 
 from pylint.extensions.docparams import DocstringParameterChecker
 from pylint.testutils import CheckerTestCase, Message, set_config
@@ -343,7 +343,7 @@ def function_foo(x, y):
         with self.assertNoMessages():
             self.checker.visit_functiondef(node)
 
-    def _visit_methods_of_class(self, node: ClassDef) -> None:
+    def _visit_methods_of_class(self, node: nodes.ClassDef) -> None:
         """Visit all methods of a class node
 
         :param node: class node