egglog 6.1.0__cp310-none-win_amd64.whl → 7.0.0__cp310-none-win_amd64.whl

egglog/thunk.py

@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Generic, TypeVar
+
+from typing_extensions import ParamSpec, TypeVarTuple, Unpack
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+
+__all__ = ["Thunk"]
+
+T = TypeVar("T")
+P = ParamSpec("P")
+TS = TypeVarTuple("TS")
+
+
+@dataclass
+class Thunk(Generic[T, Unpack[TS]]):
+    """
+    Cached delayed function call.
+    """
+
+    state: Resolved[T] | Unresolved[T, Unpack[TS]] | Resolving[T]
+
+    @classmethod
+    def fn(
+        cls, fn: Callable[[Unpack[TS]], T], *args: Unpack[TS], fallback: Callable[[], T] | None = None
+    ) -> Thunk[T, Unpack[TS]]:
+        """
+        Create a thunk based on some functions and some partial args.
+
+        If the function is called while it is being resolved recursively, will instead return the fallback, if provided.
+        """
+        return cls(Unresolved(fn, args, fallback))
+
+    @classmethod
+    def value(cls, value: T) -> Thunk[T]:
+        return Thunk(Resolved(value))
+
+    def __call__(self) -> T:
+        match self.state:
+            case Resolved(value):
+                return value
+            case Unresolved(fn, args, fallback):
+                self.state = Resolving(fallback)
+                res = fn(*args)
+                self.state = Resolved(res)
+                return res
+            case Resolving(fallback):
+                if fallback is None:
+                    msg = "Recursively resolving thunk without fallback"
+                    raise ValueError(msg)
+                return fallback()
+
+
+@dataclass
+class Resolved(Generic[T]):
+    value: T
+
+
+@dataclass
+class Unresolved(Generic[T, Unpack[TS]]):
+    fn: Callable[[Unpack[TS]], T]
+    args: tuple[Unpack[TS]]
+    fallback: Callable[[], T] | None
+
+
+@dataclass
+class Resolving(Generic[T]):
+    fallback: Callable[[], T] | None

egglog/type_constraint_solver.py

@@ -14,6 +14,7 @@ from .declarations import *
 if TYPE_CHECKING:
     from collections.abc import Collection, Iterable
 
+
 __all__ = ["TypeConstraintSolver", "TypeConstraintError"]
 
 
@@ -38,10 +39,11 @@ class TypeConstraintSolver:
         Bind the typevars of a class to the given types.
         Used for a situation like Map[int, str].create().
         """
-        cls_typevars = self._decls.get_class_decl(ref.name).type_vars
+        name = ref.name
+        cls_typevars = self._decls.get_class_decl(name).type_vars
         if len(cls_typevars) != len(ref.args):
             raise TypeConstraintError(f"Mismatch of typevars {cls_typevars} and {ref}")
-        bound_typevars = self._cls_typevar_index_to_type[ref.name]
+        bound_typevars = self._cls_typevar_index_to_type[name]
         for i, arg in enumerate(ref.args):
             bound_typevars[cls_typevars[i]] = arg
 
@@ -117,6 +119,7 @@ class TypeConstraintSolver:
                 if cls_name is None:
                     msg = "Cannot infer typevar without class name"
                     raise RuntimeError(msg)
+
                 class_typevars = self._cls_typevar_index_to_type[cls_name]
                 if typevar in class_typevars:
                     if class_typevars[typevar] != arg:

{egglog-6.1.0.dist-info → egglog-7.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: egglog
-Version: 6.1.0
+Version: 7.0.0
 Classifier: Environment :: MacOS X
 Classifier: Environment :: Win32 (MS Windows)
 Classifier: Intended Audience :: Developers
@@ -19,22 +19,6 @@ Classifier: Typing :: Typed
 Requires-Dist: typing-extensions
 Requires-Dist: black
 Requires-Dist: graphviz
-Requires-Dist: pre-commit; extra == 'dev'
-Requires-Dist: ruff; extra == 'dev'
-Requires-Dist: mypy; extra == 'dev'
-Requires-Dist: anywidget[dev]; extra == 'dev'
-Requires-Dist: egglog[docs,test]; extra == 'dev'
-Requires-Dist: scikit-learn; extra == 'array'
-Requires-Dist: array_api_compat; extra == 'array'
-Requires-Dist: numba==0.59.0rc1; extra == 'array'
-Requires-Dist: llvmlite==0.42.0rc1; extra == 'array'
-Requires-Dist: pytest; extra == 'test'
-Requires-Dist: mypy; extra == 'test'
-Requires-Dist: syrupy; extra == 'test'
-Requires-Dist: egglog[array]; extra == 'test'
-Requires-Dist: pytest-codspeed; extra == 'test'
-Requires-Dist: pytest-benchmark; extra == 'test'
-Requires-Dist: pytest-xdist; extra == 'test'
 Requires-Dist: pydata-sphinx-theme; extra == 'docs'
 Requires-Dist: myst-nb; extra == 'docs'
 Requires-Dist: sphinx-autodoc-typehints; extra == 'docs'
@@ -47,10 +31,26 @@ Requires-Dist: egglog[array]; extra == 'docs'
 Requires-Dist: line-profiler; extra == 'docs'
 Requires-Dist: sphinxcontrib-mermaid; extra == 'docs'
 Requires-Dist: ablog; extra == 'docs'
+Requires-Dist: pre-commit; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
+Requires-Dist: mypy; extra == 'dev'
+Requires-Dist: anywidget[dev]; extra == 'dev'
+Requires-Dist: egglog[docs,test]; extra == 'dev'
+Requires-Dist: pytest; extra == 'test'
+Requires-Dist: mypy; extra == 'test'
+Requires-Dist: syrupy; extra == 'test'
+Requires-Dist: egglog[array]; extra == 'test'
+Requires-Dist: pytest-codspeed; extra == 'test'
+Requires-Dist: pytest-benchmark; extra == 'test'
+Requires-Dist: pytest-xdist; extra == 'test'
+Requires-Dist: scikit-learn; extra == 'array'
+Requires-Dist: array_api_compat; extra == 'array'
+Requires-Dist: numba==0.59.0rc1; extra == 'array'
+Requires-Dist: llvmlite==0.42.0rc1; extra == 'array'
+Provides-Extra: docs
 Provides-Extra: dev
-Provides-Extra: array
 Provides-Extra: test
-Provides-Extra: docs
+Provides-Extra: array
 License-File: LICENSE
 Summary: e-graphs in Python built around the the egglog rust library
 License: MIT

{egglog-6.1.0.dist-info → egglog-7.0.0.dist-info}/RECORD

@@ -1,11 +1,13 @@
-egglog-6.1.0.dist-info/METADATA,sha256=1MhEh2rFwsmv5yqI6TZ2hoHfGwxUBSzFk1Z_slTV0HU,3838
-egglog-6.1.0.dist-info/WHEEL,sha256=keLBtIOE7ZgLyd8Cijw37iqv72a2jhfxMdDu8Unr7zo,96
-egglog-6.1.0.dist-info/license_files/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
-egglog/bindings.pyi,sha256=YPGerQlOpSO6kHyIYTUh1hnOopy4cyAKMPtljr3a1rg,11514
-egglog/builtins.py,sha256=lc7jWEjf5hUXzPRvo8pJrkGFIUBC0oLXV4U6WYXDrEw,11959
+egglog-7.0.0.dist-info/METADATA,sha256=Z_hOJtco7EDSXWxJH_L2Sod2Za3dYICl6SaSN-UKVD0,3838
+egglog-7.0.0.dist-info/WHEEL,sha256=keLBtIOE7ZgLyd8Cijw37iqv72a2jhfxMdDu8Unr7zo,96
+egglog-7.0.0.dist-info/license_files/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
+egglog/bindings.pyi,sha256=q9gyhu_kTcAPXtyu2IzHhytzjhua4sDEjM1h3XrEw60,11588
+egglog/builtins.py,sha256=lI-WAQGsXYYNvPXBnR6SpRpoTLFQ8u6GaVkrJ8nb-GY,11962
 egglog/config.py,sha256=mALVaxh7zmGrbuyzaVKVmYKcu1lF703QsKJw8AF7gSM,176
-egglog/declarations.py,sha256=JMivLe8Cwna3BsLgls9X_LU-ZQuFapyD-KkZpkf4Eq0,39062
-egglog/egraph.py,sha256=y-vPuD0U1h8ajNrjy3V5q-r61j6yDtQbeOyDFaZy9eE,69677
+egglog/conversion.py,sha256=9unbL5LYkYQ-Jc9gsWCd1VwdrI_w3mwudAcm-5PDPcc,6143
+egglog/declarations.py,sha256=zTm0TQd7YB3DfxCldONDMg-AH0leCoU4EyQ-pzfUs7A,15454
+egglog/egraph.py,sha256=HwK4vfI5eSndaK4tVinV5bWJ_K4O5Pzq8zA2un2nAcs,63151
+egglog/egraph_state.py,sha256=2UuZ2XOmTXm-D_Ytfw5LjicsjpM52vLevbOF3ucJtyc,17691
 egglog/examples/bool.py,sha256=pWZTjfXR1cFy3KcihLBU5AF5rn83ImORlhUUJ1YiAXc,733
 egglog/examples/eqsat_basic.py,sha256=ORXFYYEDsEZK2IPhHtoFsd-LdjMiQi1nn7kix4Nam0s,1011
 egglog/examples/fib.py,sha256=wAn-PjazxgHDkXAU4o2xTk_GtM_iGL0biV66vWM1st4,520
@@ -16,19 +18,22 @@ egglog/examples/README.rst,sha256=QrbfmivODBvUvmY3-dHarcbC6bEvwoqAfTDhiI-aJxU,23
 egglog/examples/resolution.py,sha256=sKkbRI_v9XkQM0DriacKLINqKKDqYGFhvMCAS9tZbTA,2203
 egglog/examples/schedule_demo.py,sha256=iJtIbcLaZ7zK8UalY0z7KAKMqYjQx0MKTsNF24lKtik,652
 egglog/examples/__init__.py,sha256=KuhaJFOyz_rpUvEqZubsgLnv6rhQNE_AVFXA6bUnpdY,34
-egglog/exp/array_api.py,sha256=ozOPzHzWBT6-5TFX4Flc9Em4I_WBywZZsekmadi06wg,40901
+egglog/exp/array_api.py,sha256=ClS1kbPq1xY4LwmllCo3wHstmZ7dbSKQP8xZcaah6fQ,41310
 egglog/exp/array_api_jit.py,sha256=HIZzd0G17u-u_F4vfRdhoYvRo-ETx5HFO3RBcOfLcco,1287
-egglog/exp/array_api_numba.py,sha256=b7pA3Mk0s0H5s0N5ZWofcWUXb4EVmnUZMEJk0EXyukw,2881
+egglog/exp/array_api_numba.py,sha256=OjONBLdFRukx4vKiNK_rBxjMzsxbWpMEdD7JcGHJjmY,2924
 egglog/exp/array_api_program_gen.py,sha256=crgUYXXNhQdfTq31FSIpWLIfzNsgQD8ngg3OosCtIgg,19680
 egglog/exp/program_gen.py,sha256=2qlfc-dVSNbPFQVl0eIzUzf-yw28AeaARa4PUC1xtRA,12257
+egglog/exp/siu_examples.py,sha256=KZmpkSCgbL4uqHhx2Jh9Adz1_cDbkIlyXPa1Kh_0O_8,784
 egglog/exp/__init__.py,sha256=G9zeKUcPBgIhgUg1meC86OfZVFETYIugyHWseTcCe_0,52
 egglog/graphviz_widget.py,sha256=YtI7LCFWHihDQ1qLvdj2SVYEcuZLSooFUYheunOTxdc,1339
 egglog/ipython_magic.py,sha256=VA19xAb6Sz7-IxlJBbnZW_gVFDqaYNnvdMB9QitndjE,1254
+egglog/pretty.py,sha256=PORBaI1jlPSgySipX8foIS5hBZPlY-W5bv2clD-J1Ps,17051
 egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-egglog/runtime.py,sha256=0Vje8UZTbrHba5d1wjKocbHiio4MDPWraAOrLQXDj9Q,27484
-egglog/type_constraint_solver.py,sha256=u-b2SZxptCxjYNcht_TQSAebIVEcf3P_-2-TJk8uaqs,5671
+egglog/runtime.py,sha256=U6fsbTU81hWZdW8chCbnuA3OoOSZ8JnWTDE-M5WF3OM,18722
+egglog/thunk.py,sha256=jsj5lp6qU87GTseYip_shE7ZTYbF9E1GDcB8R1ubQh8,1925
+egglog/type_constraint_solver.py,sha256=0c8oy-sLCVN0XndfAyRxO1AvIgfQuszw0NaRJ9ILysE,5692
 egglog/widget.css,sha256=WJS2M1wQdujhSTCakMa_ZXuoTPre1Uy1lPcvBE1LZQU,102
 egglog/widget.js,sha256=UNOER3sYZ-bS7Qhw9S6qtpR81FuHa5DzXQaWWtQq334,2021
-egglog/__init__.py,sha256=AGM-7H_Vx1Xb0Z-tc0C9Y8JbQxpa6hdYfhSU_dnT3Mw,244
-egglog/bindings.cp310-win_amd64.pyd,sha256=M6YHB0Ef-G-xSpFHmIFtPa-sL8yQ2t_C_DfDoPjCJ0s,4991488
-egglog-6.1.0.dist-info/RECORD,,
+egglog/__init__.py,sha256=iUrVe5fb0XFyMCS3CwTjwhKEtU8KpIkdTpJpnUhm8o0,247
+egglog/bindings.cp310-win_amd64.pyd,sha256=KSpjek2AtlZiqgmNTtDMuFrAn4O9FQzOXjCOuvvbH88,5108224
+egglog-7.0.0.dist-info/RECORD,,