3.0 set comprehension bug

This commit is contained in:
rocky 2023-01-16 03:50:50 -05:00
parent 50697bb79e
commit 2b3cd696db

View File

@ -465,8 +465,13 @@ class ComprehensionMixin:
self.write(": ")
self.preorder(n[1])
else:
if self.version == (3, 0) and len(n) > 1:
body = n[1]
if self.version == (3, 0):
if isinstance(n, Token):
body = store
elif len(n) > 1:
body = n[1]
else:
body = n[0]
else:
body = n[0]
self.preorder(body)