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.write(": ")
self.preorder(n[1]) self.preorder(n[1])
else: else:
if self.version == (3, 0) and len(n) > 1: if self.version == (3, 0):
body = n[1] if isinstance(n, Token):
body = store
elif len(n) > 1:
body = n[1]
else:
body = n[0]
else: else:
body = n[0] body = n[0]
self.preorder(body) self.preorder(body)