egglog 7.2.0__cp312-none-win_amd64.whl → 8.0.1__cp312-none-win_amd64.whl

egglog/visualizer_widget.py

@@ -0,0 +1,39 @@
+import pathlib
+
+import anywidget
+import traitlets
+from graphviz.backend.viewing import view
+from IPython.display import display
+from ipywidgets.embed import embed_minimal_html
+
+from .ipython_magic import IN_IPYTHON
+
+CURRENT_DIR = pathlib.Path(__file__).parent
+
+
+class VisualizerWidget(anywidget.AnyWidget):
+    """
+    Widget to render multiple graphs using the interactive visualizer.
+
+    The index will choose the one that is currently displayed, defaulting to the last one.
+    """
+
+    _esm = CURRENT_DIR / "visualizer.js"
+    _css = CURRENT_DIR / "visualizer.css"
+    egraphs = traitlets.List[str]().tag(sync=True)
+
+    def display_or_open(self) -> None:
+        """
+        Displays the widget if we are in a Jupyter environment, otherwise saves it to a file and opens it.
+        """
+        if IN_IPYTHON:
+            display(self)
+            return
+        # 1. Create a temporary html file that will stay open after close
+        # 2. Write the widget to it with embed_minimal_html
+        # 3. Open the file using the open function from graphviz
+        file = pathlib.Path.cwd() / "tmp.html"
+        # https://github.com/manzt/anywidget/issues/339#issuecomment-1755654547
+        embed_minimal_html(file, views=[self], drop_defaults=False)
+        print("Visualizer widget saved to", file)
+        view(file)

{egglog-7.2.0.dist-info → egglog-8.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: egglog
-Version: 7.2.0
+Version: 8.0.1
 Classifier: Environment :: MacOS X
 Classifier: Environment :: Win32 (MS Windows)
 Classifier: Intended Audience :: Developers
@@ -19,38 +19,39 @@ Classifier: Typing :: Typed
 Requires-Dist: typing-extensions
 Requires-Dist: black
 Requires-Dist: graphviz
-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: nbconvert; extra == 'docs'
-Requires-Dist: matplotlib; extra == 'docs'
-Requires-Dist: anywidget; extra == 'docs'
-Requires-Dist: seaborn; extra == 'docs'
-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: 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.1; extra == 'array'
-Requires-Dist: llvmlite==0.42.0; extra == 'array'
-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'
-Provides-Extra: docs
-Provides-Extra: test
+Requires-Dist: anywidget
+Requires-Dist: scikit-learn ; extra == 'array'
+Requires-Dist: array-api-compat ; extra == 'array'
+Requires-Dist: numba ==0.59.1 ; extra == 'array'
+Requires-Dist: llvmlite ==0.42.0 ; extra == 'array'
+Requires-Dist: ruff ; extra == 'dev'
+Requires-Dist: pre-commit ; 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: 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: nbconvert ; extra == 'docs'
+Requires-Dist: matplotlib ; extra == 'docs'
+Requires-Dist: anywidget ; extra == 'docs'
+Requires-Dist: seaborn ; extra == 'docs'
+Requires-Dist: egglog[array] ; extra == 'docs'
+Requires-Dist: line-profiler ; extra == 'docs'
+Requires-Dist: sphinxcontrib-mermaid ; extra == 'docs'
+Requires-Dist: ablog ; extra == 'docs'
 Provides-Extra: array
 Provides-Extra: dev
+Provides-Extra: test
+Provides-Extra: docs
 License-File: LICENSE
 Summary: e-graphs in Python built around the the egglog rust library
 License: MIT

egglog-8.0.1.dist-info/RECORD

@@ -0,0 +1,42 @@
+egglog-8.0.1.dist-info/METADATA,sha256=o6kbuHo1dQI88faQF8xLtjIR0Py8IzppNjEq6piwkoY,3884
+egglog-8.0.1.dist-info/WHEEL,sha256=VGMDu5jnp_XpyXOq-L1-cbr3x9_PzCpsKi1udyLAeSI,95
+egglog-8.0.1.dist-info/licenses/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
+egglog/bindings.pyi,sha256=Zzhm7o_UfOou_6VQw8kjsVYj8cC6VegOnuTGV0240iw,13147
+egglog/builtins.py,sha256=W42Zj8uuOiG8rGMvu1U20rIbWDGPkHHeU3qDlg9cQnU,14883
+egglog/config.py,sha256=mALVaxh7zmGrbuyzaVKVmYKcu1lF703QsKJw8AF7gSM,176
+egglog/conversion.py,sha256=Do76TJZHGhUnENCz1DNnH9gV7q4jrq6v5YTAazcglIs,7081
+egglog/declarations.py,sha256=1OZuq5hbm0_Yy8JbmJ7Sp2JXB_MQTvJ-d2Kf3kaBuCk,21265
+egglog/egraph.py,sha256=VKLaZqWXKlwO3aJ_Nayi5ZyPVdQY_Ms1xB5dELt64SQ,71917
+egglog/egraph_state.py,sha256=tcvd3bkIULnNXQbpO5nInRcGua-032epcy7J1tcE7Ww,25772
+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
+egglog/examples/higher_order_functions.py,sha256=5P-_10y8eTh-z3bze5GVzKHyFTeefTM774haF8KYGig,1164
+egglog/examples/lambda_.py,sha256=hQBOaSw_yorNcbkQVu2EhgSc0IZNWIny7asaOlcUk9s,8496
+egglog/examples/matrix.py,sha256=_zmjgfFr2O_LjTcsTD-45_38Y_M1sP3AV39K6oFxAdw,5136
+egglog/examples/ndarrays.py,sha256=T-wwef-n-3LDSjaO35zA8AZH5DXFFqq0XBSCQKEXV6E,4186
+egglog/examples/README.rst,sha256=QrbfmivODBvUvmY3-dHarcbC6bEvwoqAfTDhiI-aJxU,237
+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=svlajo4Kw0Gj1DhTt5cpg4bKMFncUK3myKnh0ATw9QQ,47776
+egglog/exp/array_api_jit.py,sha256=SEKFtx96sz4YDxKiDO1QzuT3xt_9yKa2fSLzZG_3ox8,1185
+egglog/exp/array_api_loopnest.py,sha256=oKKiLBMs1HkxbiLuScMbjiL_7l_-ZKc58CuPutxAmYg,4410
+egglog/exp/array_api_numba.py,sha256=SDYRVV45lM4tgwIbmPmOG4GIy8o4AKsUeqLqbGKolAw,2917
+egglog/exp/array_api_program_gen.py,sha256=qR6cy3O5kjgPL-5I9zZt96u5YJuxaExhfPPyWULWpjQ,21356
+egglog/exp/program_gen.py,sha256=Z2IsMMIDNb6yubn4ovkJnzqStPpNagIHS-Zca5qv300,12434
+egglog/exp/siu_examples.py,sha256=KZmpkSCgbL4uqHhx2Jh9Adz1_cDbkIlyXPa1Kh_0O_8,784
+egglog/exp/__init__.py,sha256=G9zeKUcPBgIhgUg1meC86OfZVFETYIugyHWseTcCe_0,52
+egglog/functionalize.py,sha256=6BS3BheP3NrQAbIh4dmzXDaxu746zLdYIfYTfaMvNjQ,3936
+egglog/ipython_magic.py,sha256=QDw45CA2wKi9vrBbvHZXLeMaRtoHSGLBJdqyNFtvD7o,1230
+egglog/pretty.py,sha256=Dk4H2O3vgpt_FvdxcxGDSmZEw-U4AxRKrO-89rEtWxw,21380
+egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+egglog/runtime.py,sha256=N2yQ9m9BMvekiN1R8tfy76hlrOTPsfRRLNXZZntTjLQ,23918
+egglog/thunk.py,sha256=t71q_SIRSpYh6oqOQ-TEDqMByrrwp5xvSJbVy3XONbQ,2263
+egglog/type_constraint_solver.py,sha256=MW22gwryNOqsYaB__dCCjVvdnB1Km5fovoW2IgUXaVo,5836
+egglog/visualizer.css,sha256=APwI_pG4w4OXXCc5o01TqKbwv9Fp_r-3bktM3kLIFb0,259244
+egglog/visualizer.js,sha256=wQnOcOwmUIUAmVdeZ4vLaSjZMdzdJJz-3K1AGaQ8z1E,3788815
+egglog/visualizer_widget.py,sha256=6oFfG-mHEeeh3dzng4hjU30dNI86bKEnSXismmlJTwc,1362
+egglog/__init__.py,sha256=iUrVe5fb0XFyMCS3CwTjwhKEtU8KpIkdTpJpnUhm8o0,247
+egglog/bindings.cp312-win_amd64.pyd,sha256=vfUkdGiwWW_YIh9L2eEP31w2wUgmCc3U7tjnpf-Wxyo,5003264
+egglog-8.0.1.dist-info/RECORD,,

{egglog-7.2.0.dist-info → egglog-8.0.1.dist-info}/WHEEL

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

egglog/graphviz_widget.py

@@ -1,34 +0,0 @@
-import pathlib
-
-import anywidget
-import ipywidgets
-import traitlets
-
-
-class GraphvizWidget(anywidget.AnyWidget):
-    """
-    Graphviz widget to render multiple graphviz graphs.
-
-    The index will choose the one that is currently displayed, defaulting to the last one.
-    If the index or the graphs change, there will be a re-rendering, with animation.
-    """
-
-    _esm = pathlib.Path(__file__).parent / "widget.js"
-    _css = pathlib.Path(__file__).parent / "widget.css"
-    dots = traitlets.List().tag(sync=True)
-    index = traitlets.Int(None, allow_none=True).tag(sync=True)
-    performance = traitlets.Bool(False).tag(sync=True)
-
-
-def graphviz_widget_with_slider(dots: list[str], *, performance: bool = False) -> ipywidgets.VBox:
-    n_dots = len(dots)
-    graphviz_widget = GraphvizWidget()
-    graphviz_widget.dots = dots
-    graphviz_widget.performance = performance
-    slider_widget = ipywidgets.IntSlider(max=n_dots - 1, value=0)
-    ipywidgets.jslink((slider_widget, "value"), (graphviz_widget, "index"))
-    # play_widget = ipywidgets.Play(max=n_dots - 1, repeat=True, interval=4000)
-    # ipywidgets.jslink((slider_widget, "value"), (play_widget, "value"))
-    # top = pywidgets.HBox([play_widget, slider_widget])
-    top = slider_widget
-    return ipywidgets.VBox([top, graphviz_widget])

egglog/widget.css

@@ -1,6 +0,0 @@
-.graphviz-container {
-  /* position: absolute;
-  top: 0;
-  left: 0; */
-  /* height: 100px; */
-}

egglog/widget.js

@@ -1,50 +0,0 @@
-import * as d3 from "https://cdn.skypack.dev/pin/d3@v7.8.5-eC7TKxlFLay7fmsv0gvu/dist=es2020,mode=imports,min/optimized/d3.js";
-import "https://cdn.skypack.dev/-/d3-graphviz@v5.1.0-TcGnvMu4khUzCpL7Wr2k/dist=es2020,mode=imports,min/optimized/d3-graphviz.js";
-
-export function render({ model, el }) {
-  if (!document.getElementById("graphviz_script")) {
-    const graphviz_script = document.createElement("script");
-    graphviz_script.setAttribute("id", "graphviz_script");
-    graphviz_script.setAttribute("src", "https://unpkg.com/@hpcc-js/wasm/dist/graphviz.umd.js");
-    graphviz_script.setAttribute("type", "javascript/worker");
-
-    document.head.appendChild(graphviz_script);
-  }
-  //   let getCount = () => model.get("dot");
-
-  const div = document.createElement("div");
-  div.classList.add("graphviz-container");
-  const graphContainer = d3.select(div);
-
-  let setDot = () => {
-    const dots = model.get("dots");
-    const performance = model.get("performance");
-    // Use the last dots if there is no index
-    const dot = dots[model.get("index")];
-    // const width = div.clientWidth;
-    // const height = div.clientHeight;
-    const graphviz = graphContainer
-      .graphviz({
-        // Fit graph to that size, so that all is visible
-        fit: true,
-        // Set to be as big as container
-        // width,
-        // height,
-        // Don't animate transitions between shapes for performance
-        tweenPaths: !performance,
-        tweenShapes: !performance,
-        useWorker: true,
-      })
-      .transition(() => d3.transition("t").duration(2000).ease(d3.easeLinear))
-      .renderDot(dot);
-    // If we have made a zoom selection, reset that before transitioning
-    // TODO: figure out how to transition BOTH zoom and dot at once
-    // if (graphviz._zoomSelection) {
-    //   graphviz.resetZoom();
-    // }
-  };
-
-  model.on("change:dots change:index", setDot);
-  el.appendChild(div);
-  requestAnimationFrame(() => setTimeout(setDot, 0));
-}

egglog-7.2.0.dist-info/RECORD

@@ -1,40 +0,0 @@
-egglog-7.2.0.dist-info/METADATA,sha256=O3Uq5oIIHWZ-bG2XVILykl-0dYyuNuoR9Dxj38v6bXs,3829
-egglog-7.2.0.dist-info/WHEEL,sha256=5VSD4WjrAS64a1-QM6eGcLa48TnO082fkviTtVvn8m4,96
-egglog-7.2.0.dist-info/license_files/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
-egglog/bindings.pyi,sha256=iFdtYHqPjuVt46qh1IE81cOI-lbGgPISKQB-3ERNzhI,11770
-egglog/builtins.py,sha256=p5oZLDleCgQC8G2Zp2EvmqsfVnpgmARqUHqosKSSKnQ,13120
-egglog/config.py,sha256=mALVaxh7zmGrbuyzaVKVmYKcu1lF703QsKJw8AF7gSM,176
-egglog/conversion.py,sha256=4JhocGd1_nwmFMVNCzDDwj6aWfhBFTX2hm_7Xc_DiUM,6328
-egglog/declarations.py,sha256=WpZ1yyw4vlWm3Xi_otPP2IU51IqNcDmJA5y20RcBFBQ,18583
-egglog/egraph.py,sha256=Q3ZJhtzzWcsukuarT0vT48-DkfHe17adLLUHn-pILPE,68657
-egglog/egraph_state.py,sha256=kXCNt4zenG1meRv5T6B8xI5oH_NreA06JMhmqv_F-og,21802
-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
-egglog/examples/higher_order_functions.py,sha256=CBZqnqVdYGgztHw5QWtcnUhzs1nnBcw4O-gjSzAxjRc,1203
-egglog/examples/lambda_.py,sha256=hQBOaSw_yorNcbkQVu2EhgSc0IZNWIny7asaOlcUk9s,8496
-egglog/examples/matrix.py,sha256=_zmjgfFr2O_LjTcsTD-45_38Y_M1sP3AV39K6oFxAdw,5136
-egglog/examples/ndarrays.py,sha256=T-wwef-n-3LDSjaO35zA8AZH5DXFFqq0XBSCQKEXV6E,4186
-egglog/examples/README.rst,sha256=QrbfmivODBvUvmY3-dHarcbC6bEvwoqAfTDhiI-aJxU,237
-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=b7MoUuH2sIW3KnUX7fGPaflwlCCskgrt1GsidVT-KG4,41474
-egglog/exp/array_api_jit.py,sha256=HIZzd0G17u-u_F4vfRdhoYvRo-ETx5HFO3RBcOfLcco,1287
-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=s5XBIB7kMrxV25aO-F4NW5Jli2DOoc72wwswFYxJPjM,19561
-egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-egglog/runtime.py,sha256=s32xdoD_mIdDRi_PJQ5E_kSNyhUkT310osH38kZfCSk,23174
-egglog/thunk.py,sha256=bm1zZUzwey0DijtMZkSulyfiWjCAg_1UmXaSG8qvvjs,2170
-egglog/type_constraint_solver.py,sha256=MW22gwryNOqsYaB__dCCjVvdnB1Km5fovoW2IgUXaVo,5836
-egglog/widget.css,sha256=WJS2M1wQdujhSTCakMa_ZXuoTPre1Uy1lPcvBE1LZQU,102
-egglog/widget.js,sha256=UNOER3sYZ-bS7Qhw9S6qtpR81FuHa5DzXQaWWtQq334,2021
-egglog/__init__.py,sha256=iUrVe5fb0XFyMCS3CwTjwhKEtU8KpIkdTpJpnUhm8o0,247
-egglog/bindings.cp312-win_amd64.pyd,sha256=fYGMMgJx6iehBvLFrjqCRropokuwQPKBMcs5E4dSxG4,4620288
-egglog-7.2.0.dist-info/RECORD,,