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

whitequark · whitequark · commit 89d137f9bd9f · 2015-08-27T16:36:23.000-05:00
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)
 
