diff --git a/vm/src/builtins/code.rs b/vm/src/builtins/code.rs index 24cb970077..6e4c71e12f 100644 --- a/vm/src/builtins/code.rs +++ b/vm/src/builtins/code.rs @@ -218,6 +218,18 @@ impl PyRef { self.code.source_path.to_owned() } + #[pygetset] + pub fn co_cellvars(self, vm: &VirtualMachine) -> PyTupleRef { + let cellvars = self + .code + .cellvars + .deref() + .iter() + .map(|name| name.to_pyobject(vm)) + .collect(); + vm.ctx.new_tuple(cellvars) + } + #[pygetset] fn co_firstlineno(self) -> usize { self.code.first_line_number