Subsume lambda functions after replacing

saulshanabrook · saulshanabrook · commit 2adc98a36237 · 2024-12-06T12:58:48.000-05:00
diff --git a/docs/changelog.md b/docs/changelog.md
@@ -8,6 +8,7 @@ _This project uses semantic versioning_
 - Add support for subsuming rewrite generated by default function and method definitions
 - Add better error message when using @function in class (thanks @shinawy)
 - Add error method if `@method` decorator is in wrong place
+- Subsumes lambda functions after replacing
 
 ## 8.0.1 (2024-10-24)
 
diff --git a/python/egglog/builtins.py b/python/egglog/builtins.py
@@ -532,7 +532,8 @@ def _convert_function(a: FunctionType) -> UnstableFn:
     transformed_fn = functionalize(a, value_to_annotation)
     assert isinstance(transformed_fn, partial)
     return UnstableFn(
-        function(ruleset=get_current_ruleset(), use_body_as_name=True)(transformed_fn.func), *transformed_fn.args
+        function(ruleset=get_current_ruleset(), use_body_as_name=True, subsume=True)(transformed_fn.func),
+        *transformed_fn.args,
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -532,7 +532,8 @@ def _convert_function(a: FunctionType) -> UnstableFn:`
`532`	`532`	`transformed_fn = functionalize(a, value_to_annotation)`
`533`	`533`	`assert isinstance(transformed_fn, partial)`
`534`	`534`	`return UnstableFn(`
`535`		`- function(ruleset=get_current_ruleset(), use_body_as_name=True)(transformed_fn.func), *transformed_fn.args`
	`535`	`+ function(ruleset=get_current_ruleset(), use_body_as_name=True, subsume=True)(transformed_fn.func),`
	`536`	`+ *transformed_fn.args,`
`536`	`537`	`)`
`537`	`538`
`538`	`539`