Fixes

python/egglog/egraph.py

@@ -677,7 +677,7 @@ class EGraph(_BaseModule):
     _decl_stack: list[Declarations] = field(default_factory=list, repr=False)
     _token: Optional[Token[EGraph]] = None
 
-    def __post_init__(self, modules: list[Module], seminaive) -> None:
+    def __post_init__(self, modules: list[Module], seminaive: bool) -> None:  # type: ignore
         super().__post_init__(modules)
         self._egraph = bindings.EGraph(seminaive=seminaive)
         for m in self._flatted_deps:
@@ -899,7 +899,7 @@ def inner(*__args: PyObject, __fn_locals=fn_locals) -> PyObject:
                 new_kvs.append(arg)
                 eval_str += f"__arg_{i}, "
             eval_str += ")"
-            return py_eval(eval_str, fn_locals.dict_update(new_kvs), fn_globals)
+            return py_eval(eval_str, fn_locals.dict_update(*new_kvs), fn_globals)
 
         return inner
 

python/egglog/examples/lambda.py

@@ -2,7 +2,6 @@
 Lambda Calculus
 ===============
 """
-# mypy: disable-error-code=empty-body
 from __future__ import annotations
 
 from typing import Callable, ClassVar

python/egglog/examples/ndarrays.py

@@ -4,7 +4,6 @@
 
 Example of building NDarray in the vein of Mathemetics of Arrays.
 """
-# mypy: disable-error-code=empty-body
 from __future__ import annotations
 
 from egglog import *

python/egglog/exp/array_api.py

@@ -1,4 +1,3 @@
-# mypy: disable-error-code=empty-body
 from __future__ import annotations
 
 import itertools

python/tests/test_high_level.py

@@ -1,4 +1,3 @@
-# mypy: disable-error-code="empty-body"
 from __future__ import annotations
 
 import importlib
@@ -8,7 +7,6 @@
 
 import pytest
 from egglog import *
-from egglog.builtins import count_matches
 from egglog.declarations import (
     CallDecl,
     FunctionRef,