diff options
Diffstat (limited to 'astroid/transforms.py')
-rw-r--r-- | astroid/transforms.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/astroid/transforms.py b/astroid/transforms.py index 840c1b11..f6c72794 100644 --- a/astroid/transforms.py +++ b/astroid/transforms.py @@ -145,10 +145,10 @@ class TransformVisitor: """Unregister the given transform.""" self.transforms[node_class].remove((transform, predicate)) # type: ignore[index, arg-type] - def visit(self, module: nodes.NodeNG) -> SuccessfulInferenceResult: + def visit(self, node: nodes.NodeNG) -> SuccessfulInferenceResult: """Walk the given astroid *tree* and transform each encountered node. Only the nodes which have transforms registered will actually be replaced or changed. """ - return self._visit(module) + return self._visit(node) |