From 2b3cd696db0fb9d7e2540c78c284ee39322bd0bb Mon Sep 17 00:00:00 2001 From: rocky Date: Mon, 16 Jan 2023 03:50:50 -0500 Subject: [PATCH] 3.0 set comprehension bug --- uncompyle6/semantics/gencomp.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/uncompyle6/semantics/gencomp.py b/uncompyle6/semantics/gencomp.py index 78d8677d..c1018c90 100644 --- a/uncompyle6/semantics/gencomp.py +++ b/uncompyle6/semantics/gencomp.py @@ -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)