1.1 --- a/compiler/ast.py Thu May 17 20:22:53 2012 +0200
1.2 +++ b/compiler/ast.py Fri May 18 23:35:21 2012 +0200
1.3 @@ -1054,8 +1054,8 @@
1.4 def __repr__(self):
1.5 return "SetComp(%s, %s)" % (repr(self.expr), repr(self.quals))
1.6
1.7 - def visit(self, visitor):
1.8 - return visitor.visitSetComp(self)
1.9 + def visit(self, visitor, *args):
1.10 + return visitor.visitSetComp(self, *args)
1.11
1.12 class DictComp(Node):
1.13 def __init__(self, key, value, quals, lineno=None):
1.14 @@ -1081,8 +1081,8 @@
1.15 def __repr__(self):
1.16 return "DictComp(%s, %s, %s)" % (repr(self.key), repr(self.value), repr(self.quals))
1.17
1.18 - def visit(self, visitor):
1.19 - return visitor.visitDictComp(self)
1.20 + def visit(self, visitor, *args):
1.21 + return visitor.visitDictComp(self, *args)
1.22
1.23 class Mod(Node):
1.24 def __init__(self, leftright, lineno=None):
1.25 @@ -1356,8 +1356,8 @@
1.26 def __repr__(self):
1.27 return "Set(%s)" % (repr(self.nodes),)
1.28
1.29 - def visit(self, visitor):
1.30 - return visitor.visitSet(self)
1.31 + def visit(self, visitor, *args):
1.32 + return visitor.visitSet(self, *args)
1.33
1.34 class Slice(Node):
1.35 def __init__(self, expr, flags, lower, upper, lineno=None):