Update codegen.rs

GreasySlug · Dec 28, 2023 · a128719 · a128719
1 parent 93f1dc0
commit a128719
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/crates/erg_compiler/codegen.rs b/crates/erg_compiler/codegen.rs
@@ -781,6 +781,12 @@ impl PyCodeGenerator {
             | "invert" | "is_" | "is_not" | "call" => {
                 self.load_operators();
             }
+            "CodeType" => {
+                self.emit_global_import_items(
+                    Identifier::public("types"),
+                    vec![(Identifier::public("CodeType"), None)],
+                );
+            }
             // NoneType is not defined in the global scope, use `type(None)` instead
             "NoneType" => {
                 self.emit_push_null();