Fix notebook bug

Author: saulshanabrook

python/egglog/egraph.py

@@ -809,6 +809,7 @@ def _add_default_rewrite(
     resolved_value = resolve_literal(type_ref, default_rewrite, Thunk.value(decls))
     rewrite_decl = DefaultRewriteDecl(ref, resolved_value.__egg_typed_expr__.expr, subsume)
     ruleset_decls = _add_default_rewrite_inner(decls, rewrite_decl, ruleset)
+    ruleset_decls |= decls
     ruleset_decls |= resolved_value
 
 

python/tests/test_array_api.py

@@ -373,6 +373,15 @@ def lda(X: NDArray, y: NDArray):
     return run_lda(X, y)
 
 
+def test_lda_symbolic_build_cold_schedule():
+    X_arr = NDArray.var("X")
+    y_arr = NDArray.var("y")
+    egraph = EGraph()
+    with set_array_api_egraph(egraph):
+        res = lda(X_arr, y_arr)
+    assert isinstance(res, NDArray)
+
+
 @pytest.mark.parametrize(
     "program",
     [

python/tests/test_high_level.py

@@ -658,6 +658,19 @@ def f() -> A:
 
         check_eq(f(), A(), r)
 
+    def test_function_ruleset_can_run_after_materialization_without_registration(self):
+        r = ruleset()
+
+        @function(ruleset=r)
+        def f() -> A:
+            return A()
+
+        # Materialize the function once so its default rewrite is added to the ruleset,
+        # but do not register any expression that would separately add `f` to the egraph.
+        f()
+        egraph = EGraph()
+        assert not egraph.run(r).updated
+
     def test_constant(self):
         a = constant("a", A, A())
         check_eq(a, A(), run())