round_trace: fix the detection of arguments

my2to3/fixes/fix_round_trace.py

@@ -29,13 +29,13 @@ class FixRoundTrace(BaseDynamicTraceFix):
 
   def transform(self, node, results):
     args = results['args']
-    if args.type == lib2to3.pgen2.token.NUMBER:
-      # e.g. round(1.23)
+    # e.g. round(1.23, 4)
+    if args.type == syms.arglist:
+      children = [leaf.clone() for leaf in args.children]
+    # e.g. round(1.23)
+    else:
       children = [
         args.clone()
       ]
-    else:
-      # e.g. round(1,23, 4)
-      children = [leaf.clone() for leaf in args.children]
 
     node.replace(self.traced_call(node, children))