Skip to content

Commit 2b3cd69

Browse files
committed
3.0 set comprehension bug
1 parent 50697bb commit 2b3cd69

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

uncompyle6/semantics/gencomp.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -465,8 +465,13 @@ def comprehension_walk_newer(
465465
self.write(": ")
466466
self.preorder(n[1])
467467
else:
468-
if self.version == (3, 0) and len(n) > 1:
469-
body = n[1]
468+
if self.version == (3, 0):
469+
if isinstance(n, Token):
470+
body = store
471+
elif len(n) > 1:
472+
body = n[1]
473+
else:
474+
body = n[0]
470475
else:
471476
body = n[0]
472477
self.preorder(body)

0 commit comments

Comments
 (0)