Fix rewrite fast locals on 3.12 for #548

nucleic · Apr 15, 2024 · 0b34df4 · 0b34df4
1 parent bf0ea6f
commit 0b34df4
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/enaml/core/code_generator.py b/enaml/core/code_generator.py
@@ -588,10 +588,14 @@ def rewrite_to_fast_locals(self, local_names):
         stored_names = set()
         code_ops = self.code_ops
         for idx, instr in enumerate(code_ops):
+            if not isinstance(instr, bc.Instr):
+                continue
             if instr.name == "STORE_NAME":
                 stored_names.add(instr.arg)
                 instr.name = "STORE_FAST"
         for idx, instr in enumerate(code_ops):
+            if not isinstance(instr, bc.Instr):
+                continue
             i_name = instr.name
             if i_name == "LOAD_NAME":
                 i_arg = instr.arg

diff --git a/tests/test_template.py b/tests/test_template.py
@@ -241,6 +241,24 @@ def test_syntax_20():
     compile_source(source, 'Main')
 
 
+def test_syntax_21():
+    source = dedent("""\
+    template Other(*Args):
+        Slider:
+            pass
+
+    template Main(Model):
+        const args = [
+            m for m in Model.members().values()
+            if isinstance(m, Float)
+        ]
+        Other(args): *all:
+            pass
+
+    """)
+    compile_source(source, 'Main')
+
+
 #------------------------------------------------------------------------------
 # Bad Template Syntax
 #------------------------------------------------------------------------------