egglog 0.4.0__cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

egglog/type_constraint_solver.py

@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from itertools import chain, repeat
+from typing import Collection, Optional
+
+from .declarations import *
+
+__all__ = ["TypeConstraintSolver", "TypeConstraintError"]
+
+
+class TypeConstraintError(RuntimeError):
+    pass
+
+
+@dataclass
+class TypeConstraintSolver:
+    """
+    Given some typevars and types, solves the constraints to resolve the typevars.
+    """
+
+    # Mapping of class typevar index to their types
+    _cls_typevar_index_to_type: dict[int, JustTypeRef] = field(default_factory=dict)
+
+    @classmethod
+    def from_type_parameters(cls, tps: Collection[JustTypeRef]) -> TypeConstraintSolver:
+        """
+        Create a TypeInference from a parameterized class, given as a bound typevar.
+
+        Used for a situation like Map[int, str].create()
+        """
+        cs = cls()
+        # For each param in the class, use this as the ith bound class typevar
+        for i, tp in enumerate(tps):
+            cs._cls_typevar_index_to_type[i] = tp
+        return cs
+
+    def infer_return_type(
+        self,
+        fn_args: Collection[TypeOrVarRef],
+        fn_return: TypeOrVarRef,
+        fn_var_args: Optional[TypeOrVarRef],
+        args: Collection[JustTypeRef],
+    ) -> JustTypeRef:
+        # Infer the type of each type variable based on the actual types of the arguments
+        self._infer_typevars_zip(fn_args, fn_var_args, args)
+        # Substitute the type variables with their inferred types
+        return self._subtitute_typevars(fn_return)
+
+    def _infer_typevars_zip(
+        self, fn_args: Collection[TypeOrVarRef], fn_var_args: Optional[TypeOrVarRef], args: Collection[JustTypeRef]
+    ) -> None:
+        if len(fn_args) != len(args) if fn_var_args is None else len(fn_args) > len(args):
+            raise TypeConstraintError(f"Expected {len(fn_args)} args, got {len(args)}")
+        all_fn_args = fn_args if fn_var_args is None else chain(fn_args, repeat(fn_var_args))
+        for fn_arg, arg in zip(all_fn_args, args):
+            self._infer_typevars(fn_arg, arg)
+
+    def _infer_typevars(self, fn_arg: TypeOrVarRef, arg: JustTypeRef) -> None:
+        if isinstance(fn_arg, TypeRefWithVars):
+            if fn_arg.name != arg.name:
+                raise TypeConstraintError(f"Expected {fn_arg.name}, got {arg.name}")
+            self._infer_typevars_zip(fn_arg.args, None, arg.args)
+        elif fn_arg.index not in self._cls_typevar_index_to_type:
+            self._cls_typevar_index_to_type[fn_arg.index] = arg
+        elif self._cls_typevar_index_to_type[fn_arg.index] != arg:
+            raise TypeConstraintError(f"Expected {fn_arg}, got {arg}")
+
+    def _subtitute_typevars(self, tp: TypeOrVarRef) -> JustTypeRef:
+        if isinstance(tp, ClassTypeVarRef):
+            try:
+                return self._cls_typevar_index_to_type[tp.index]
+            except KeyError:
+                raise TypeConstraintError(f"Not enough bound typevars for {tp}")
+        elif isinstance(tp, TypeRefWithVars):
+            return JustTypeRef(
+                tp.name,
+                tuple(self._subtitute_typevars(arg) for arg in tp.args),
+            )

egglog-0.4.0.dist-info/METADATA

@@ -0,0 +1,53 @@
+Metadata-Version: 2.1
+Name: egglog
+Version: 0.4.0
+Classifier: Environment :: MacOS X
+Classifier: Environment :: Win32 (MS Windows)
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Rust
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Classifier: Topic :: Software Development :: Compilers
+Classifier: Topic :: Software Development :: Interpreters
+Classifier: Typing :: Typed
+Requires-Dist: typing-extensions
+Requires-Dist: black
+Requires-Dist: graphviz
+Requires-Dist: pytest; extra == 'test'
+Requires-Dist: mypy; extra == 'test'
+Requires-Dist: pydata-sphinx-theme; extra == 'docs'
+Requires-Dist: myst-nb; extra == 'docs'
+Requires-Dist: sphinx-autodoc-typehints; extra == 'docs'
+Requires-Dist: sphinx-gallery; extra == 'docs'
+Requires-Dist: matplotlib; extra == 'docs'
+Requires-Dist: nbconvert; extra == 'docs'
+Requires-Dist: pre-commit; extra == 'dev'
+Requires-Dist: black; extra == 'dev'
+Requires-Dist: mypy; extra == 'dev'
+Requires-Dist: flake8; extra == 'dev'
+Requires-Dist: isort; extra == 'dev'
+Provides-Extra: test
+Provides-Extra: docs
+Provides-Extra: dev
+License-File: LICENSE
+Summary: e-graphs in Python built around the the egglog rust library
+License: MIT
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+
+# `egglog` Python wrapper
+
+[![Documentation Status](https://readthedocs.org/projects/egg-smol-python/badge/?version=latest)](https://egg-smol-python.readthedocs.io/en/latest/?badge=latest) [![Test](https://github.com/metadsl/egglog-python/actions/workflows/CI.yml/badge.svg?branch=main)](https://github.com/metadsl/egglog-python/actions/workflows/CI.yml) [![PyPi Package](https://img.shields.io/pypi/v/egglog.svg)](https://pypi.org/project/egglog/) [![License](https://img.shields.io/pypi/l/egglog.svg)](https://pypi.org/project/egglog/) [![Python Versions](https://img.shields.io/pypi/pyversions/egglog.svg)](https://pypi.org/project/egglog/) [![pre-commit](https://img.shields.io/badge/pre--commit-enabled-brightgreen?logo=pre-commit)](https://github.com/pre-commit/pre-commit)
+
+`egglog` is a Python package that provides bindings to the Rust library [`egglog`](https://github.com/egraphs-good/egglog/),
+allowing you to use e-graphs in Python for optimization, symbolic computation, and analysis.
+
+Please see the [documentation](https://egg-smol-python.readthedocs.io/en/latest/?badge=latest) for more information.
+

egglog-0.4.0.dist-info/RECORD

@@ -0,0 +1,25 @@
+egglog-0.4.0.dist-info/METADATA,sha256=YaRlVy3sIsMgI8zUkrlQYtLkkuXkWgtVKNi4RTGZ5Qg,2840
+egglog-0.4.0.dist-info/WHEEL,sha256=cMZaAELfCq-Wny4Kd-Zb88nk1QVHKOQ56uOMKlU1ooA,130
+egglog-0.4.0.dist-info/license_files/LICENSE,sha256=9cYb25jNLLm6mgX6Io4aYZ2UaiWcSCl_ROMIev2vX-Y,1064
+egglog/__init__.py,sha256=y9IIN--blzi8SjA7MUuN8WiVW8Mpl-dRffbh2to8oHA,143
+egglog/bindings.pyi,sha256=kS4m5XQ7kR9-2U5jGX66G2FsifTGEtVXGSr_Tx8Y1GU,8188
+egglog/examples/resolution.py,sha256=l_YkhZUenncGhp8JkQQ6upLgAaTlcIrmr3crrn22Sog,2100
+egglog/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+egglog/examples/lambda.py,sha256=pA1vGapQ--80AlUUr4r9gT_wW1pQh5h8Kz7wiCFzjM0,8643
+egglog/examples/README.rst,sha256=ztTvpofR0eotSqGoCy_C1fPLDPCncjvcqDanXtLHNNU,232
+egglog/examples/eqsat_basic.py,sha256=ezv6PlJRAySfW5KBl90B_ZdqGXxzeO-aFLD6wTfxPfM,990
+egglog/examples/matrix.py,sha256=XYsAO1oXYIVl-h4KxHkVsLs4sdIIEKAdI6h3kg_gMow,5281
+egglog/examples/ndarrays.py,sha256=zvLaWmqP-85PbVFx3XDcGMJ_04DKHpqe3mgqxki6EFE,4230
+egglog/examples/schedule_demo.py,sha256=C4w9cevqmQbiJT_Ms2RV1vXtsvckur4J-mMAooMJG2Y,723
+egglog/examples/fib.py,sha256=IR67Ehd0tMn9efE9UTLFvuGe2h6D0Ip2X2RxT1xk6is,505
+egglog/type_constraint_solver.py,sha256=o0KOmJQhXNkuXcxmi8cTfCNsYQ0ZE5rHQK5rsXlq-R4,3153
+egglog/runtime.py,sha256=FN-hXLdOflh-tkTUOgkeX3NxznKxEpS2qRRV8rFyfl4,11975
+egglog/egraph.py,sha256=xbfvcz8nwa2xfST0G-1yJY5QHWOefS28K9CETh37MuE,36861
+egglog/declarations.py,sha256=L3hqbirdUSdfDK3rqS1fsHDeUl8D4hlKcA7kgoilkEI,32042
+egglog/builtins.py,sha256=A28JPrH5Q3iL1-hyE96zbEWI4QpYZzndKisopN9yJWU,9347
+egglog/config.py,sha256=yM3FIcVCKnhWZmHD0pxkzx7ah7t5PxZx3WUqKtA9tjU,168
+egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+egglog/ipython_magic.py,sha256=7VEOYg90X0xRkQdqXGSJZoGYv59Re-Z4Ss3l0OI1t3A,1168
+egglog/monkeypatch.py,sha256=HKabvIfRoVQoIXLR4smtU0R9Fyc4GGuWF3al5xeh65w,1140
+egglog/bindings.cpython-312-x86_64-linux-gnu.so,sha256=rE1Ln6PtZMZcHcEOENK5xXqqxKWyuieA-E8VlYedKBw,9310512
+egglog-0.4.0.dist-info/RECORD,,

egglog-0.4.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: maturin (0.13.7)
+Root-Is-Purelib: false
+Tag: cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64

egglog-0.4.0.dist-info/license_files/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 metadsl
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.