egglog 10.0.2__cp312-cp312-win_amd64.whl → 11.1.0__cp312-cp312-win_amd64.whl

{egglog-10.0.2.dist-info → egglog-11.1.0.dist-info}/RECORD

@@ -1,45 +1,45 @@
-egglog-10.0.2.dist-info/METADATA,sha256=xR0HB5r6onjZhIk878L992umIt2QZT3zzRRCEN1NkF0,4019
-egglog-10.0.2.dist-info/WHEEL,sha256=2Rq0eWWH7u9Ffm_ZQEcE2_DVE8if9XSfMophnE-xWmc,96
-egglog-10.0.2.dist-info/licenses/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
-egglog/__init__.py,sha256=TZ3IMVKoz5Qr97VFe-YT7oyT2NACikTis21PpVlSCmg,326
-egglog/bindings.cp312-win_amd64.pyd,sha256=5QoF0a8dXYh-usqUNV2SKAETYdFE8ItjyO-wP1-xxP0,4541952
-egglog/bindings.pyi,sha256=fnRiGyHuM-m0fDe91kFryUC09U87f5XXPiY6Lno8bvk,14497
-egglog/builtins.py,sha256=su57iYmxpWNt14S9Dw1CgBfSBUYiTLCS6h1ksShZUMk,30691
+egglog-11.1.0.dist-info/METADATA,sha256=865-TFtrwCy1OfpeqBU2BcK96tl3G4hG1AMb4fVNCZU,4019
+egglog-11.1.0.dist-info/WHEEL,sha256=HF3aUMilrtO42xS_fBzOVaPE8OtiNjP_RotLatII7HM,96
+egglog-11.1.0.dist-info/licenses/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
+egglog/__init__.py,sha256=CAD9lfaUAcqSoTtprQfsFYqoOcqxAYsGBDU7laaYj0w,357
+egglog/bindings.cp312-win_amd64.pyd,sha256=RBang7LLIidBOyvOFYKpCJ_QqgisOXyB0FzjsFkR5bE,10308096
+egglog/bindings.pyi,sha256=-sm1qF7dDGEsFGDbny8LEIYTFMnlDfUV7HfQ-P81Crg,14301
+egglog/builtins.py,sha256=StODsE0YzfDmRnvt3lJChwUa9TQ-M16EAGQMjVS2njI,31337
 egglog/config.py,sha256=mALVaxh7zmGrbuyzaVKVmYKcu1lF703QsKJw8AF7gSM,176
-egglog/conversion.py,sha256=o6iE32taKz-PBX8aDo5i4jGELdEai-wrzMEZUC1lqsU,10257
-egglog/declarations.py,sha256=7tT7RXHEKVoZtm14hT-Va_1jKuwOMPGFWg8VdRYhT88,23015
-egglog/egraph.py,sha256=BnLyNXJ6UqMIM2gqYBgeMXvqjbX5OAwnMTRudOUyxtU,64497
-egglog/egraph_state.py,sha256=wkuHoXmEzcUFspvNl8ZG7_xS1J7XZTK3RPnlZEuvk9k,28720
+egglog/conversion.py,sha256=YSkLn744-Ci_GmlcZAvLwmJNN8WZyYDjPI8Sqj5TN-0,11524
+egglog/declarations.py,sha256=nXO13l2s8K-PB2d4ARoIS63deuxo0RWG9zAFKG3APBA,26688
+egglog/deconstruct.py,sha256=BzcrC1Qc_hlFhfQ2kkh1JB9Ur9Vi6IyjqM6KLhrZR58,5808
+egglog/egraph.py,sha256=aI0POepFzToRW8-OZIlMY-ELinK3DvB3clBbxvPit-s,62400
+egglog/egraph_state.py,sha256=hdjM10qpeeUypNM7OREPT9uxVD0kvn6WaoMkXq96TpU,28993
 egglog/examples/README.rst,sha256=QrbfmivODBvUvmY3-dHarcbC6bEvwoqAfTDhiI-aJxU,237
 egglog/examples/__init__.py,sha256=KuhaJFOyz_rpUvEqZubsgLnv6rhQNE_AVFXA6bUnpdY,34
-egglog/examples/bignum.py,sha256=r4UeVwpKK2iSZiT5MK4qqTWhWu76CAtSJ5YGDboO9tk,568
+egglog/examples/bignum.py,sha256=KSt7V__O8IFbfvjkwOwJ5Dnw2JUFnYSyEfhfu8M71YY,567
 egglog/examples/bool.py,sha256=pWZTjfXR1cFy3KcihLBU5AF5rn83ImORlhUUJ1YiAXc,733
 egglog/examples/eqsat_basic.py,sha256=ivYHFD52sjYyl5rWQ60ScxmvcbABYUBlPobDMu6HvN4,913
 egglog/examples/fib.py,sha256=wAn-PjazxgHDkXAU4o2xTk_GtM_iGL0biV66vWM1st4,520
 egglog/examples/higher_order_functions.py,sha256=z8HgCi057k7Rt6Ms4xDz_tsmVj_Czx8JMUSWqd0HlvM,1105
 egglog/examples/lambda_.py,sha256=y0HrgtmF2Ml2h2crRvqd1fYRtJGPP_n91o-fiWfyCLM,8481
 egglog/examples/matrix.py,sha256=_zmjgfFr2O_LjTcsTD-45_38Y_M1sP3AV39K6oFxAdw,5136
-egglog/examples/multiset.py,sha256=kSUFiN4qs8WXvexPCecqk3o5DzdywXacDmIiC5IDKCM,1507
+egglog/examples/multiset.py,sha256=uAy7RZRzU2OBeNp0R4xWxZmF7NXDFGRr_dIbH26tUsY,1505
 egglog/examples/ndarrays.py,sha256=T-wwef-n-3LDSjaO35zA8AZH5DXFFqq0XBSCQKEXV6E,4186
 egglog/examples/resolution.py,sha256=sKkbRI_v9XkQM0DriacKLINqKKDqYGFhvMCAS9tZbTA,2203
 egglog/examples/schedule_demo.py,sha256=iJtIbcLaZ7zK8UalY0z7KAKMqYjQx0MKTsNF24lKtik,652
 egglog/exp/__init__.py,sha256=G9zeKUcPBgIhgUg1meC86OfZVFETYIugyHWseTcCe_0,52
-egglog/exp/array_api.py,sha256=0bqgi2nigiW5uLEa35JBOh9hFM5-oRMGYpoknzS60Jw,65828
-egglog/exp/array_api_jit.py,sha256=c0lN-411M9LmEz8uM0ELXTtfUPMJDZFmy81-x3ck6PY,1591
+egglog/exp/array_api.py,sha256=3f_lh6vI-vvxlKW3ocV8FT3FH4TQ1VqSAB6vZEnF0Fk,67567
+egglog/exp/array_api_jit.py,sha256=1mp6s2e2ySqulfb5pFuTQZ-IQ3UdDmT7eJvJME1X5hs,1731
 egglog/exp/array_api_loopnest.py,sha256=1KZ8kWtv00TVCbzqzWG92vz8Y1pBlPOLRIrJfxyDs9s,2518
 egglog/exp/array_api_numba.py,sha256=g0Yb7auB2MtJHgSzn7iVnsxn6ffL4FHggKvbEuuDwZs,2765
-egglog/exp/array_api_program_gen.py,sha256=qXaaW_Koz4P-KmQcyRHY1NDLCJL3O7NnE6iq2NX4yZU,22252
-egglog/exp/program_gen.py,sha256=obqPocwY9LpmXyjIRtONvafkJm84HKWDsR_fME0gns4,13438
+egglog/exp/array_api_program_gen.py,sha256=LatOqKiGVzpUhWIGtoZw9lLZdK3Oeg7XjR4ILLGby10,22289
+egglog/exp/program_gen.py,sha256=CiLgPo2YoQi83DAVmR7gU61ocWkxnYEV1toie8y8ppQ,13458
 egglog/exp/siu_examples.py,sha256=nPV1IaA2NfIczpJPg3l8diUhoyio02KQUu-O55Pj1EQ,751
-egglog/functionalize.py,sha256=7yBi5rlYS56lP4Fz_2REyOyj56Tag5R80bXkUXfj5pk,3938
 egglog/ipython_magic.py,sha256=QDw45CA2wKi9vrBbvHZXLeMaRtoHSGLBJdqyNFtvD7o,1230
-egglog/pretty.py,sha256=8ErTOWGHDY96W5B1ekctKNwRm0fqPHUrqLlX-h1fSJM,20986
+egglog/pretty.py,sha256=pSTOMAgywLZ_0Y3oxYPmIbtxsVADWFw60G3LKTOYFWk,20632
 egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-egglog/runtime.py,sha256=zxcgUR5N_-qHs12vcH4r0oANY-9ra16jwlCAsNxZyC4,26465
+egglog/runtime.py,sha256=pOuAnqK_5Fb3Q0tfjC0JUIPoi9eEQtQBg38TtmnNSUQ,28664
 egglog/thunk.py,sha256=97JH5VVrr5m3yaua40StfDsQYEvACXirBzZv9CHHLbY,2368
 egglog/type_constraint_solver.py,sha256=uRLho3KQUtMn3ncugXADN5SfNqmqekduOl5CuWhhx4M,4631
-egglog/version_compat.py,sha256=axYHagkkErdrW1aW6OyWyHUqjpdKNZQ9N4U1KpL4x3s,3595
+egglog/version_compat.py,sha256=QGgZo85mwGsuFqNkRfLiGdxGR0K-B5Pax0f7fRjMW50,3599
 egglog/visualizer.css,sha256=DDmxZfLmFygwVkQwel0gGUiM5BZqgunawWeSfaRYBr8,259399
 egglog/visualizer.js,sha256=kDkYzE5cLLhmpNaWqSfQWzD0SlSsNe9KzT_GpNqhWgo,3789092
 egglog/visualizer_widget.py,sha256=TGkf61P_f4EvCFcyqwQC_8XYUkujFhcQnEG9vHWLPC4,1358
-egglog-10.0.2.dist-info/RECORD,,
+egglog-11.1.0.dist-info/RECORD,,

{egglog-10.0.2.dist-info → egglog-11.1.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.7)
+Generator: maturin (1.9.3)
 Root-Is-Purelib: false
 Tag: cp312-cp312-win_amd64

egglog/functionalize.py

@@ -1,91 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Callable
-from functools import partial
-from inspect import Parameter, signature
-from typing import Any, TypeVar, cast
-
-__all__ = ["functionalize"]
-
-
-T = TypeVar("T", bound=Callable)
-
-
-# TODO: Add `to_lift` param so that we only transform those with vars in them to args
-
-
-def functionalize(f: T, get_annotation: Callable[[object], type | None]) -> T:
-    """
-    Takes a function and returns a new function with all names (co_names) and free variables (co_freevars) added as arguments
-    and then partially applied with their values. The second arg, get_annotation, will be applied to all values
-    to get their type annotation. If it is None, that arg will not be added as a parameter.
-
-    For example if you have:
-
-        def get_annotation(x): return int if x <= 10 else None
-
-        g = 10
-        def f(a, a2):
-            def h(b: Z):
-                return a + a2 + b + g
-
-            return functionalize(h, get_annotation)
-        res = f(9, 11)
-
-    It should be equivalent to (according to body, signature, and annotations) (Note that the new arguments will be positional only):
-
-        def h(a: get_annotation(a), g: get_annotation(g), b: Z):
-            return a + b + g
-        res = partial(h, a, g)
-    """
-    code = f.__code__
-    names = tuple(n for n in code.co_names if n in f.__globals__)
-    free_vars = code.co_freevars
-
-    global_values: list[Any] = [f.__globals__[name] for name in names]
-    free_var_values = [cell.cell_contents for cell in f.__closure__] if f.__closure__ else []
-    assert len(free_var_values) == len(free_vars), "Free vars and their values do not match"
-    global_values_filtered = [
-        (i, name, value, annotation)
-        for i, (name, value) in enumerate(zip(names, global_values, strict=True))
-        if (annotation := get_annotation(value)) is not None
-    ]
-    free_var_values_filtered = [
-        (i, name, value, annotation)
-        for i, (name, value) in enumerate(zip(free_vars, free_var_values, strict=True))
-        if (annotation := get_annotation(value)) is not None
-    ]
-    additional_arg_filtered = global_values_filtered + free_var_values_filtered
-
-    # Create a wrapper function
-    def wrapper(*args):
-        # Split args into names, free vars and other args
-        name_args, free_var_args, rest_args = (
-            args[: (n_names := len(global_values_filtered))],
-            args[n_names : (n_args := len(additional_arg_filtered))],
-            args[n_args:],
-        )
-        # Update globals with names
-        f.__globals__.update({
-            name: arg for (_, name, _, _), arg in zip(global_values_filtered, name_args, strict=False)
-        })
-        # update function free vars with free var args
-        for (i, _, _, _), value in zip(free_var_values_filtered, free_var_args, strict=True):
-            assert f.__closure__, "Function does not have closure"
-            f.__closure__[i].cell_contents = value
-        return f(*rest_args)
-
-    # Set the signature of the new function to a signature with the free vars and names added as arguments
-    orig_signature = signature(f)
-    wrapper.__signature__ = orig_signature.replace(  # type: ignore[attr-defined]
-        parameters=[
-            *[Parameter(n, Parameter.POSITIONAL_OR_KEYWORD) for _, n, _, _ in additional_arg_filtered],
-            *orig_signature.parameters.values(),
-        ]
-    )
-    # Set the annotations of the new function to the annotations of the original function + annotations of passed in values
-    wrapper.__annotations__ = f.__annotations__ | {n: a for _, n, _, a in additional_arg_filtered}
-    wrapper.__name__ = f.__name__
-
-    # Partially apply the wrapper function with the current values of the free vars
-    return cast("T", partial(wrapper, *(v for _, _, v, _ in additional_arg_filtered)))