fixes with statement with multiple items

Closes #1478
2020-10-06 15:19:45 +08:00 · 2020-10-06 15:19:45 +08:00 · 88d346fa26
parent 9214e0f3e2
commit 88d346fa26
1 changed files with 2 additions and 2 deletions
--- a/artiq/compiler/transforms/artiq_ir_generator.py
+++ b/artiq/compiler/transforms/artiq_ir_generator.py
@ -883,8 +883,8 @@ class ARTIQIRGenerator(algorithm.Visitor):
                    self.current_assign = None

                none = self.append(ir.Alloc([], builtins.TNone()))
-                cleanup.append(lambda:
-                    self._user_call(exit_fn, [none, none, none], {}))
+                call = self._user_call(exit_fn, [none, none, none], {})
+                cleanup.append(lambda: call)

        self._try_finally(
            body_gen=lambda: self.visit(node.body),