SymbolTableType::Lambda (#5942)

youknowone · web-flow · commit 8b6c78c884ca · 2025-07-11T13:35:52.000+09:00
diff --git a/compiler/codegen/src/symboltable.rs b/compiler/codegen/src/symboltable.rs
@@ -80,6 +80,7 @@ pub enum SymbolTableType {
     Module,
     Class,
     Function,
+    Lambda,
     Comprehension,
     TypeParams,
 }
@@ -90,6 +91,7 @@ impl fmt::Display for SymbolTableType {
             Self::Module => write!(f, "module"),
             Self::Class => write!(f, "class"),
             Self::Function => write!(f, "function"),
+            Self::Lambda => write!(f, "lambda"),
             Self::Comprehension => write!(f, "comprehension"),
             Self::TypeParams => write!(f, "type parameter"),
             // TODO missing types from the C implementation
@@ -493,7 +495,7 @@ impl SymbolTableAnalyzer {
                     location: None,
                 });
             }
-            SymbolTableType::Function => {
+            SymbolTableType::Function | SymbolTableType::Lambda => {
                 if let Some(parent_symbol) = symbols.get_mut(&symbol.name) {
                     if let SymbolScope::Unknown = parent_symbol.scope {
                         // this information is new, as the assignment is done in inner scope
@@ -1140,7 +1142,7 @@ impl SymbolTableBuilder<'_> {
                 } else {
                     self.enter_scope(
                         "lambda",
-                        SymbolTableType::Function,
+                        SymbolTableType::Lambda,
                         self.line_index_start(expression.range()),
                     );
                 }
diff --git a/compiler/core/src/bytecode.rs b/compiler/core/src/bytecode.rs
@@ -414,7 +414,7 @@ op_arg_enum!(
         // PrepReraiseS tar = 1,
         // TypeVarWithBound = 2,
         // TypeVarWithConstraints = 3,
-        // SetFunctionTypeParams = 4,
+        SetFunctionTypeParams = 4,
         /// Set default value for type parameter (PEP 695)
         SetTypeparamDefault = 5,
     }
diff --git a/vm/src/frame.rs b/vm/src/frame.rs
@@ -2286,6 +2286,12 @@ impl ExecutingFrame<'_> {
             bytecode::IntrinsicFunction2::SetTypeparamDefault => {
                 crate::stdlib::typing::set_typeparam_default(arg1, arg2, vm)
             }
+            bytecode::IntrinsicFunction2::SetFunctionTypeParams => {
+                // arg1 is the function, arg2 is the type params tuple
+                // Set __type_params__ attribute on the function
+                arg1.set_attr("__type_params__", arg2, vm)?;
+                Ok(arg1)
+            }
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -414,7 +414,7 @@ op_arg_enum!(`
`414`	`414`	`// PrepReraiseS tar = 1,`
`415`	`415`	`// TypeVarWithBound = 2,`
`416`	`416`	`// TypeVarWithConstraints = 3,`
`417`		`- // SetFunctionTypeParams = 4,`
	`417`	`+ SetFunctionTypeParams = 4,`
`418`	`418`	`/// Set default value for type parameter (PEP 695)`
`419`	`419`	`SetTypeparamDefault = 5,`
`420`	`420`	`}`
Original file line number	Diff line number	Diff line change
`@@ -2286,6 +2286,12 @@ impl ExecutingFrame<'_> {`
`2286`	`2286`	`bytecode::IntrinsicFunction2::SetTypeparamDefault => {`
`2287`	`2287`	`crate::stdlib::typing::set_typeparam_default(arg1, arg2, vm)`
`2288`	`2288`	`}`
	`2289`	`+ bytecode::IntrinsicFunction2::SetFunctionTypeParams => {`
	`2290`	`+ // arg1 is the function, arg2 is the type params tuple`
	`2291`	`+ // Set __type_params__ attribute on the function`
	`2292`	`+ arg1.set_attr("__type_params__", arg2, vm)?;`
	`2293`	`+ Ok(arg1)`
	`2294`	`+ }`
`2289`	`2295`	`}`
`2290`	`2296`	`}`
`2291`	`2297`