Merge fix-lookup-value into polynomials

Author: saulshanabrook

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "egglog_python"
-version = "12.0.0"
+version = "13.0.1"
 edition = "2024"
 
 

Cargo.toml

@@ -0,0 +1,4 @@
+{
+  "url": "https://context7.com/egraphs-good/egglog-python",
+  "public_key": "pk_OgxDfp3XyhQai6fY0aQTI"
+}

context7.json

@@ -4,7 +4,14 @@ _This project uses semantic versioning_
 
 ## UNRELEASED
 
-- Support using facts as union actions, add conversions to multisets, and update multiset examlpe [#382](https://github.com/egraphs-good/egglog-python/pull/382)
+## 13.0.1 (2026-03-04)
+
+- Fix install by adding cloudpickle as required dependency [#405](https://github.com/egraphs-good/egglog-python/pull/405)
+- Add __classdictcell__ to ignored attributes (Python 3.14 support) [#403](https://github.com/egraphs-good/egglog-python/pull/403)
+## 13.0.0 (2026-03-03)
+
+- Support using facts as union actions, add conversions to multisets, and update multiset example [#382](https://github.com/egraphs-good/egglog-python/pull/382)
+- Fix lookup of cost model based on value (see [zulip for issue](https://egraphs.zulipchat.com/#narrow/channel/375765-egg.2Fegglog/topic/Cost.20function.3A.20using.20function.20values.20of.20subtrees/near/577062352))
 
 ## 12.0.0 (2025-11-16)
 

docs/changelog.md

@@ -26,7 +26,7 @@ classifiers = [
     "Topic :: Software Development :: Interpreters",
     "Typing :: Typed",
 ]
-dependencies = ["typing-extensions", "black", "graphviz", "anywidget"]
+dependencies = ["typing-extensions", "black", "graphviz", "anywidget", "cloudpickle>=3"]
 
 [project.optional-dependencies]
 
@@ -35,8 +35,7 @@ array = [
     "array_api_compat",
     "numba>=0.59.1",
     "llvmlite>=0.42.0",
-    "numpy>2",
-    "cloudpickle>=3",
+    "numpy>2"
 ]
 dev = [
     "ruff",

pyproject.toml

@@ -130,6 +130,9 @@
     "__firstlineno__",
     "__static_attributes__",
     "__match_args__",
+    # Added in 3.14
+    # https://egraphs.zulipchat.com/#narrow/channel/375765-egg.2Fegglog/topic/Cost.20function.3A.20using.20function.20values.20of.20subtrees/near/577236488
+    "__classdictcell__",
     # Ignore all reflected binary method
     *(f"__r{m[2:]}" for m in NUMERIC_BINARY_METHODS),
 }

python/egglog/egraph.py

@@ -680,6 +680,8 @@ def _generate_callable_egg_name(self, ref: CallableRef) -> str:
                 assert_never(ref)
 
     def typed_expr_to_value(self, typed_expr: TypedExprDecl) -> bindings.Value:
+        if isinstance(typed_expr.expr, ValueDecl):
+            return typed_expr.expr.value
         egg_expr = self.typed_expr_to_egg(typed_expr, False)
         return self.egraph.eval_expr(egg_expr)[1]
 

python/egglog/egraph_state.py

@@ -1568,3 +1568,37 @@ def __radd__(self, other: object) -> tuple[X, X]: ...
 
     assert X(1) + 10 == (X(1), 10)
     assert 10 + X(1) == (X(10), X(1))
+
+
+def test_custom_cost_model_size():
+    """
+    https://egraphs.zulipchat.com/#narrow/channel/375765-egg.2Fegglog/topic/Cost.20function.3A.20using.20function.20values.20of.20subtrees/near/577062352
+    """
+
+    class KAT(Expr):
+        @classmethod
+        def eps(cls) -> KAT: ...
+
+        @classmethod
+        def emp(cls) -> KAT: ...
+
+        def func(self, other: KAT) -> KAT: ...
+
+        def size(self) -> i64: ...
+
+    eps, emp = KAT.eps(), KAT.emp()
+
+    eg = EGraph()
+    q0 = eg.let("q0", KAT.func(eps, emp))
+
+    eg.register(set_(eps.size()).to(i64(1)))
+    eg.register(set_(emp.size()).to(i64(0)))
+
+    def conv_cost(eg, expr, child_costs):
+        if isinstance(expr, KAT):
+            args = get_callable_args(expr)
+            return sum(int(eg.lookup_function_value(cast("KAT", a).size())) for a in args)
+
+        return 2
+
+    assert eg.extract(q0, include_cost=True, cost_model=conv_cost) == (KAT.eps().func(KAT.emp()), 1)