algorithm.Visitor.visit: return a value if an array is passed as well.

m-labs · Aug 27, 2015 · 89d137f · 89d137f
1 parent 6d74932
commit 89d137f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pythonparser/algorithm.py b/pythonparser/algorithm.py
@@ -36,7 +36,7 @@ def _visit_one(self, node):
     def visit(self, obj):
         """Visit a node or a list of nodes. Other values are ignored"""
         if isinstance(obj, list):
-            [self.visit(elt) for elt in obj]
+            return [self.visit(elt) for elt in obj]
         elif isinstance(obj, ast.AST):
             return self._visit_one(obj)