egglog 6.0.0__cp312-none-win_amd64.whl → 6.1.0__cp312-none-win_amd64.whl

egglog/builtins.py

@@ -35,20 +35,17 @@ __all__ = [
 
 
 class String(Expr, builtin=True):
-    def __init__(self, value: str) -> None:
-        ...
+    def __init__(self, value: str) -> None: ...
 
     @method(egg_fn="replace")
-    def replace(self, old: StringLike, new: StringLike) -> String:
-        ...
+    def replace(self, old: StringLike, new: StringLike) -> String: ...
 
 
 StringLike: TypeAlias = String | str
 
 
 @function(egg_fn="+", builtin=True)
-def join(*strings: StringLike) -> String:
-    ...
+def join(*strings: StringLike) -> String: ...
 
 
 converter(str, String, String)
@@ -57,28 +54,22 @@ BoolLike = Union["Bool", bool]
 
 
 class Bool(Expr, egg_sort="bool", builtin=True):
-    def __init__(self, value: bool) -> None:
-        ...
+    def __init__(self, value: bool) -> None: ...
 
     @method(egg_fn="not")
-    def __invert__(self) -> Bool:
-        ...
+    def __invert__(self) -> Bool: ...
 
     @method(egg_fn="and")
-    def __and__(self, other: BoolLike) -> Bool:
-        ...
+    def __and__(self, other: BoolLike) -> Bool: ...
 
     @method(egg_fn="or")
-    def __or__(self, other: BoolLike) -> Bool:
-        ...
+    def __or__(self, other: BoolLike) -> Bool: ...
 
     @method(egg_fn="xor")
-    def __xor__(self, other: BoolLike) -> Bool:
-        ...
+    def __xor__(self, other: BoolLike) -> Bool: ...
 
     @method(egg_fn="=>")
-    def implies(self, other: BoolLike) -> Bool:
-        ...
+    def implies(self, other: BoolLike) -> Bool: ...
 
 
 converter(bool, Bool, Bool)
@@ -88,218 +79,167 @@ i64Like = Union["i64", int]  # noqa: N816
 
 
 class i64(Expr, builtin=True):  # noqa: N801
-    def __init__(self, value: int) -> None:
-        ...
+    def __init__(self, value: int) -> None: ...
 
     @method(egg_fn="+")
-    def __add__(self, other: i64Like) -> i64:
-        ...
+    def __add__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="-")
-    def __sub__(self, other: i64Like) -> i64:
-        ...
+    def __sub__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="*")
-    def __mul__(self, other: i64Like) -> i64:
-        ...
+    def __mul__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="/")
-    def __truediv__(self, other: i64Like) -> i64:
-        ...
+    def __truediv__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="%")
-    def __mod__(self, other: i64Like) -> i64:
-        ...
+    def __mod__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="&")
-    def __and__(self, other: i64Like) -> i64:
-        ...
+    def __and__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="|")
-    def __or__(self, other: i64Like) -> i64:
-        ...
+    def __or__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="^")
-    def __xor__(self, other: i64Like) -> i64:
-        ...
+    def __xor__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="<<")
-    def __lshift__(self, other: i64Like) -> i64:
-        ...
+    def __lshift__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn=">>")
-    def __rshift__(self, other: i64Like) -> i64:
-        ...
+    def __rshift__(self, other: i64Like) -> i64: ...
 
-    def __radd__(self, other: i64Like) -> i64:
-        ...
+    def __radd__(self, other: i64Like) -> i64: ...
 
-    def __rsub__(self, other: i64Like) -> i64:
-        ...
+    def __rsub__(self, other: i64Like) -> i64: ...
 
-    def __rmul__(self, other: i64Like) -> i64:
-        ...
+    def __rmul__(self, other: i64Like) -> i64: ...
 
-    def __rtruediv__(self, other: i64Like) -> i64:
-        ...
+    def __rtruediv__(self, other: i64Like) -> i64: ...
 
-    def __rmod__(self, other: i64Like) -> i64:
-        ...
+    def __rmod__(self, other: i64Like) -> i64: ...
 
-    def __rand__(self, other: i64Like) -> i64:
-        ...
+    def __rand__(self, other: i64Like) -> i64: ...
 
-    def __ror__(self, other: i64Like) -> i64:
-        ...
+    def __ror__(self, other: i64Like) -> i64: ...
 
-    def __rxor__(self, other: i64Like) -> i64:
-        ...
+    def __rxor__(self, other: i64Like) -> i64: ...
 
-    def __rlshift__(self, other: i64Like) -> i64:
-        ...
+    def __rlshift__(self, other: i64Like) -> i64: ...
 
-    def __rrshift__(self, other: i64Like) -> i64:
-        ...
+    def __rrshift__(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="not-i64")
-    def __invert__(self) -> i64:
-        ...
+    def __invert__(self) -> i64: ...
 
     @method(egg_fn="<")
     def __lt__(self, other: i64Like) -> Unit:  # type: ignore[empty-body,has-type]
         ...
 
     @method(egg_fn=">")
-    def __gt__(self, other: i64Like) -> Unit:
-        ...
+    def __gt__(self, other: i64Like) -> Unit: ...
 
     @method(egg_fn="<=")
     def __le__(self, other: i64Like) -> Unit:  # type: ignore[empty-body,has-type]
         ...
 
     @method(egg_fn=">=")
-    def __ge__(self, other: i64Like) -> Unit:
-        ...
+    def __ge__(self, other: i64Like) -> Unit: ...
 
     @method(egg_fn="min")
-    def min(self, other: i64Like) -> i64:
-        ...
+    def min(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="max")
-    def max(self, other: i64Like) -> i64:
-        ...
+    def max(self, other: i64Like) -> i64: ...
 
     @method(egg_fn="to-string")
-    def to_string(self) -> String:
-        ...
+    def to_string(self) -> String: ...
 
     @method(egg_fn="bool-<")
-    def bool_lt(self, other: i64Like) -> Bool:
-        ...
+    def bool_lt(self, other: i64Like) -> Bool: ...
 
     @method(egg_fn="bool->")
-    def bool_gt(self, other: i64Like) -> Bool:
-        ...
+    def bool_gt(self, other: i64Like) -> Bool: ...
 
     @method(egg_fn="bool-<=")
-    def bool_le(self, other: i64Like) -> Bool:
-        ...
+    def bool_le(self, other: i64Like) -> Bool: ...
 
     @method(egg_fn="bool->=")
-    def bool_ge(self, other: i64Like) -> Bool:
-        ...
+    def bool_ge(self, other: i64Like) -> Bool: ...
 
 
 converter(int, i64, i64)
 
 
 @function(builtin=True, egg_fn="count-matches")
-def count_matches(s: StringLike, pattern: StringLike) -> i64:
-    ...
+def count_matches(s: StringLike, pattern: StringLike) -> i64: ...
 
 
 f64Like = Union["f64", float]  # noqa: N816
 
 
 class f64(Expr, builtin=True):  # noqa: N801
-    def __init__(self, value: float) -> None:
-        ...
+    def __init__(self, value: float) -> None: ...
 
     @method(egg_fn="neg")
-    def __neg__(self) -> f64:
-        ...
+    def __neg__(self) -> f64: ...
 
     @method(egg_fn="+")
-    def __add__(self, other: f64Like) -> f64:
-        ...
+    def __add__(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="-")
-    def __sub__(self, other: f64Like) -> f64:
-        ...
+    def __sub__(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="*")
-    def __mul__(self, other: f64Like) -> f64:
-        ...
+    def __mul__(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="/")
-    def __truediv__(self, other: f64Like) -> f64:
-        ...
+    def __truediv__(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="%")
-    def __mod__(self, other: f64Like) -> f64:
-        ...
+    def __mod__(self, other: f64Like) -> f64: ...
 
-    def __radd__(self, other: f64Like) -> f64:
-        ...
+    def __radd__(self, other: f64Like) -> f64: ...
 
-    def __rsub__(self, other: f64Like) -> f64:
-        ...
+    def __rsub__(self, other: f64Like) -> f64: ...
 
-    def __rmul__(self, other: f64Like) -> f64:
-        ...
+    def __rmul__(self, other: f64Like) -> f64: ...
 
-    def __rtruediv__(self, other: f64Like) -> f64:
-        ...
+    def __rtruediv__(self, other: f64Like) -> f64: ...
 
-    def __rmod__(self, other: f64Like) -> f64:
-        ...
+    def __rmod__(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="<")
     def __lt__(self, other: f64Like) -> Unit:  # type: ignore[empty-body,has-type]
         ...
 
     @method(egg_fn=">")
-    def __gt__(self, other: f64Like) -> Unit:
-        ...
+    def __gt__(self, other: f64Like) -> Unit: ...
 
     @method(egg_fn="<=")
     def __le__(self, other: f64Like) -> Unit:  # type: ignore[empty-body,has-type]
         ...
 
     @method(egg_fn=">=")
-    def __ge__(self, other: f64Like) -> Unit:
-        ...
+    def __ge__(self, other: f64Like) -> Unit: ...
 
     @method(egg_fn="min")
-    def min(self, other: f64Like) -> f64:
-        ...
+    def min(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="max")
-    def max(self, other: f64Like) -> f64:
-        ...
+    def max(self, other: f64Like) -> f64: ...
 
     @method(egg_fn="to-i64")
-    def to_i64(self) -> i64:
-        ...
+    def to_i64(self) -> i64: ...
 
     @method(egg_fn="to-f64")
     @classmethod
-    def from_i64(cls, i: i64) -> f64:
-        ...
+    def from_i64(cls, i: i64) -> f64: ...
 
     @method(egg_fn="to-string")
-    def to_string(self) -> String:
-        ...
+    def to_string(self) -> String: ...
 
 
 converter(float, f64, f64)
@@ -312,243 +252,188 @@ V = TypeVar("V", bound=Expr)
 class Map(Expr, Generic[T, V], builtin=True):
     @method(egg_fn="map-empty")
     @classmethod
-    def empty(cls) -> Map[T, V]:
-        ...
+    def empty(cls) -> Map[T, V]: ...
 
     @method(egg_fn="map-insert")
-    def insert(self, key: T, value: V) -> Map[T, V]:
-        ...
+    def insert(self, key: T, value: V) -> Map[T, V]: ...
 
     @method(egg_fn="map-get")
-    def __getitem__(self, key: T) -> V:
-        ...
+    def __getitem__(self, key: T) -> V: ...
 
     @method(egg_fn="map-not-contains")
-    def not_contains(self, key: T) -> Unit:
-        ...
+    def not_contains(self, key: T) -> Unit: ...
 
     @method(egg_fn="map-contains")
-    def contains(self, key: T) -> Unit:
-        ...
+    def contains(self, key: T) -> Unit: ...
 
     @method(egg_fn="map-remove")
-    def remove(self, key: T) -> Map[T, V]:
-        ...
+    def remove(self, key: T) -> Map[T, V]: ...
 
     @method(egg_fn="rebuild")
-    def rebuild(self) -> Map[T, V]:
-        ...
+    def rebuild(self) -> Map[T, V]: ...
 
 
 class Set(Expr, Generic[T], builtin=True):
     @method(egg_fn="set-of")
-    def __init__(self, *args: T) -> None:
-        ...
+    def __init__(self, *args: T) -> None: ...
 
     @method(egg_fn="set-empty")
     @classmethod
-    def empty(cls) -> Set[T]:
-        ...
+    def empty(cls) -> Set[T]: ...
 
     @method(egg_fn="set-insert")
-    def insert(self, value: T) -> Set[T]:
-        ...
+    def insert(self, value: T) -> Set[T]: ...
 
     @method(egg_fn="set-not-contains")
-    def not_contains(self, value: T) -> Unit:
-        ...
+    def not_contains(self, value: T) -> Unit: ...
 
     @method(egg_fn="set-contains")
-    def contains(self, value: T) -> Unit:
-        ...
+    def contains(self, value: T) -> Unit: ...
 
     @method(egg_fn="set-remove")
-    def remove(self, value: T) -> Set[T]:
-        ...
+    def remove(self, value: T) -> Set[T]: ...
 
     @method(egg_fn="set-union")
-    def __or__(self, other: Set[T]) -> Set[T]:
-        ...
+    def __or__(self, other: Set[T]) -> Set[T]: ...
 
     @method(egg_fn="set-diff")
-    def __sub__(self, other: Set[T]) -> Set[T]:
-        ...
+    def __sub__(self, other: Set[T]) -> Set[T]: ...
 
     @method(egg_fn="set-intersect")
-    def __and__(self, other: Set[T]) -> Set[T]:
-        ...
+    def __and__(self, other: Set[T]) -> Set[T]: ...
 
     @method(egg_fn="rebuild")
-    def rebuild(self) -> Set[T]:
-        ...
+    def rebuild(self) -> Set[T]: ...
 
 
 class Rational(Expr, builtin=True):
     @method(egg_fn="rational")
-    def __init__(self, num: i64Like, den: i64Like) -> None:
-        ...
+    def __init__(self, num: i64Like, den: i64Like) -> None: ...
 
     @method(egg_fn="to-f64")
-    def to_f64(self) -> f64:
-        ...
+    def to_f64(self) -> f64: ...
 
     @method(egg_fn="+")
-    def __add__(self, other: Rational) -> Rational:
-        ...
+    def __add__(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="-")
-    def __sub__(self, other: Rational) -> Rational:
-        ...
+    def __sub__(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="*")
-    def __mul__(self, other: Rational) -> Rational:
-        ...
+    def __mul__(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="/")
-    def __truediv__(self, other: Rational) -> Rational:
-        ...
+    def __truediv__(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="min")
-    def min(self, other: Rational) -> Rational:
-        ...
+    def min(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="max")
-    def max(self, other: Rational) -> Rational:
-        ...
+    def max(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="neg")
-    def __neg__(self) -> Rational:
-        ...
+    def __neg__(self) -> Rational: ...
 
     @method(egg_fn="abs")
-    def __abs__(self) -> Rational:
-        ...
+    def __abs__(self) -> Rational: ...
 
     @method(egg_fn="floor")
-    def floor(self) -> Rational:
-        ...
+    def floor(self) -> Rational: ...
 
     @method(egg_fn="ceil")
-    def ceil(self) -> Rational:
-        ...
+    def ceil(self) -> Rational: ...
 
     @method(egg_fn="round")
-    def round(self) -> Rational:
-        ...
+    def round(self) -> Rational: ...
 
     @method(egg_fn="pow")
-    def __pow__(self, other: Rational) -> Rational:
-        ...
+    def __pow__(self, other: Rational) -> Rational: ...
 
     @method(egg_fn="log")
-    def log(self) -> Rational:
-        ...
+    def log(self) -> Rational: ...
 
     @method(egg_fn="sqrt")
-    def sqrt(self) -> Rational:
-        ...
+    def sqrt(self) -> Rational: ...
 
     @method(egg_fn="cbrt")
-    def cbrt(self) -> Rational:
-        ...
+    def cbrt(self) -> Rational: ...
 
     @method(egg_fn="numer")  # type: ignore[misc]
     @property
-    def numer(self) -> i64:
-        ...
+    def numer(self) -> i64: ...
 
     @method(egg_fn="denom")  # type: ignore[misc]
     @property
-    def denom(self) -> i64:
-        ...
+    def denom(self) -> i64: ...
 
 
 class Vec(Expr, Generic[T], builtin=True):
     @method(egg_fn="vec-of")
-    def __init__(self, *args: T) -> None:
-        ...
+    def __init__(self, *args: T) -> None: ...
 
     @method(egg_fn="vec-empty")
     @classmethod
-    def empty(cls) -> Vec[T]:
-        ...
+    def empty(cls) -> Vec[T]: ...
 
     @method(egg_fn="vec-append")
-    def append(self, *others: Vec[T]) -> Vec[T]:
-        ...
+    def append(self, *others: Vec[T]) -> Vec[T]: ...
 
     @method(egg_fn="vec-push")
-    def push(self, value: T) -> Vec[T]:
-        ...
+    def push(self, value: T) -> Vec[T]: ...
 
     @method(egg_fn="vec-pop")
-    def pop(self) -> Vec[T]:
-        ...
+    def pop(self) -> Vec[T]: ...
 
     @method(egg_fn="vec-not-contains")
-    def not_contains(self, value: T) -> Unit:
-        ...
+    def not_contains(self, value: T) -> Unit: ...
 
     @method(egg_fn="vec-contains")
-    def contains(self, value: T) -> Unit:
-        ...
+    def contains(self, value: T) -> Unit: ...
 
     @method(egg_fn="vec-length")
-    def length(self) -> i64:
-        ...
+    def length(self) -> i64: ...
 
     @method(egg_fn="vec-get")
-    def __getitem__(self, index: i64Like) -> T:
-        ...
+    def __getitem__(self, index: i64Like) -> T: ...
 
     @method(egg_fn="rebuild")
-    def rebuild(self) -> Vec[T]:
-        ...
+    def rebuild(self) -> Vec[T]: ...
 
 
 class PyObject(Expr, builtin=True):
-    def __init__(self, value: object) -> None:
-        ...
+    def __init__(self, value: object) -> None: ...
 
     @method(egg_fn="py-from-string")
     @classmethod
-    def from_string(cls, s: StringLike) -> PyObject:
-        ...
+    def from_string(cls, s: StringLike) -> PyObject: ...
 
     @method(egg_fn="py-to-string")
-    def to_string(self) -> String:
-        ...
+    def to_string(self) -> String: ...
 
     @method(egg_fn="py-to-bool")
-    def to_bool(self) -> Bool:
-        ...
+    def to_bool(self) -> Bool: ...
 
     @method(egg_fn="py-dict-update")
-    def dict_update(self, *keys_and_values: object) -> PyObject:
-        ...
+    def dict_update(self, *keys_and_values: object) -> PyObject: ...
 
     @method(egg_fn="py-from-int")
     @classmethod
-    def from_int(cls, i: i64Like) -> PyObject:
-        ...
+    def from_int(cls, i: i64Like) -> PyObject: ...
 
     @method(egg_fn="py-dict")
     @classmethod
-    def dict(cls, *keys_and_values: object) -> PyObject:
-        ...
+    def dict(cls, *keys_and_values: object) -> PyObject: ...
 
 
 converter(object, PyObject, PyObject)
 
 
 @function(builtin=True, egg_fn="py-eval")
-def py_eval(code: StringLike, globals: object = PyObject.dict(), locals: object = PyObject.dict()) -> PyObject:
-    ...
+def py_eval(code: StringLike, globals: object = PyObject.dict(), locals: object = PyObject.dict()) -> PyObject: ...
 
 
 class PyObjectFunction(Protocol):
-    def __call__(self, *__args: PyObject) -> PyObject:
-        ...
+    def __call__(self, *__args: PyObject) -> PyObject: ...
 
 
 def py_eval_fn(fn: Callable) -> PyObjectFunction:
@@ -563,8 +448,7 @@ def py_eval_fn(fn: Callable) -> PyObjectFunction:
         new_kvs: list[object] = []
         eval_str = "__fn("
         for i, arg in enumerate(__args):
-            new_kvs.append(f"__arg_{i}")
-            new_kvs.append(arg)
+            new_kvs.extend((f"__arg_{i}", arg))
             eval_str += f"__arg_{i}, "
         eval_str += ")"
         return py_eval(eval_str, PyObject({"__fn": __fn}).dict_update(*new_kvs), __fn.__globals__)

egglog/declarations.py

@@ -103,8 +103,7 @@ UNARY_METHODS = {
 @runtime_checkable
 class HasDeclerations(Protocol):
     @property
-    def __egg_decls__(self) -> Declarations:
-        ...
+    def __egg_decls__(self) -> Declarations: ...
 
 
 DeclerationsLike: TypeAlias = Union[HasDeclerations, None, "Declarations"]
@@ -1000,6 +999,16 @@ class TypedExprDecl:
     def to_egg(self, decls: Declarations) -> bindings._Expr:
         return self.expr.to_egg(decls)
 
+    def descendants(self) -> list[TypedExprDecl]:
+        """
+        Returns a list of all the descendants of this expression.
+        """
+        l = [self]
+        if isinstance(self.expr, CallDecl):
+            for a in self.expr.args:
+                l.extend(a.descendants())
+        return l
+
 
 @dataclass
 class ClassDecl: