egglog 7.1.0__cp310-none-win_amd64.whl → 8.0.0__cp310-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.1.0.dist-info → egglog-8.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: egglog
-Version: 7.1.0
+Version: 8.0.0
 Classifier: Environment :: MacOS X
 Classifier: Environment :: Win32 (MS Windows)
 Classifier: Intended Audience :: Developers
@@ -19,34 +19,35 @@ Classifier: Typing :: Typed
 Requires-Dist: typing-extensions
 Requires-Dist: black
 Requires-Dist: graphviz
-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'
-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'
+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: 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: 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

egglog-8.0.0.dist-info/RECORD

@@ -0,0 +1,42 @@
+egglog-8.0.0.dist-info/METADATA,sha256=eh-TWX0GzeUHB9G_aqlqRChViH29Czcv5D99LuDD4kQ,3884
+egglog-8.0.0.dist-info/WHEEL,sha256=g5HWqh_ET1t00jeNp7yG4_MnCmGwbeYGzUw6BhdEuM0,95
+egglog-8.0.0.dist-info/licenses/LICENSE,sha256=TfaboMVZ81Q6OUaKjU7z6uVjSlcGKclLYcOpgDbm9_s,1091
+egglog/bindings.pyi,sha256=n48ha4m7lm2rRr7dbotGjH4rY5tNPrd3yrVzS9oN1ys,12749
+egglog/builtins.py,sha256=W42Zj8uuOiG8rGMvu1U20rIbWDGPkHHeU3qDlg9cQnU,14883
+egglog/config.py,sha256=mALVaxh7zmGrbuyzaVKVmYKcu1lF703QsKJw8AF7gSM,176
+egglog/conversion.py,sha256=vvxT8GBT5yaczV7GXtClBQ9xx2C-ytpSpCnmZDChDTM,7077
+egglog/declarations.py,sha256=-wTiWYbbIvOkpKLy33j4_iOhyaOnWm5zcUk55CmfIZw,21288
+egglog/egraph.py,sha256=ZkrSTFJeCUFsjCNTZS4syAtQeVCcj8qiXX1VaYslmCM,71283
+egglog/egraph_state.py,sha256=Cc6ScY8Xv50HJmAcydQzDVzYe8th21vs9QqL2vpjrqE,25495
+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=9SPJnTuMErxxuS-1GvV9GgfBP3X5wPnmhD3_h9SvAsY,1180
+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=ytqyeIspq_hdhvcdOGSR2cOTIokLtqBiomd12lCKmBU,47615
+egglog/exp/array_api_jit.py,sha256=4NWEzGDRCebzKGJgXqZoctnFAs899PE0e4kZQqFHrlE,1181
+egglog/exp/array_api_loopnest.py,sha256=lLL9MsN6_lQzpLI1wMFbVchQzk59fn-EWU9B06Q2i9g,4305
+egglog/exp/array_api_numba.py,sha256=SDYRVV45lM4tgwIbmPmOG4GIy8o4AKsUeqLqbGKolAw,2917
+egglog/exp/array_api_program_gen.py,sha256=FWZW_4FZkv_lbhEkuuQpogGu89cE9TJ6uRKYXhfLz40,21271
+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/functionalize.py,sha256=6BS3BheP3NrQAbIh4dmzXDaxu746zLdYIfYTfaMvNjQ,3936
+egglog/ipython_magic.py,sha256=VA19xAb6Sz7-IxlJBbnZW_gVFDqaYNnvdMB9QitndjE,1254
+egglog/pretty.py,sha256=MiNonL4_vublshR8R88534KuAxEU983QFEPkOXM7mog,21356
+egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+egglog/runtime.py,sha256=L3h5g4e4LUzG-psewYmtDNmPkXfbZuZBJGgAJ4RNl6g,23918
+egglog/thunk.py,sha256=t71q_SIRSpYh6oqOQ-TEDqMByrrwp5xvSJbVy3XONbQ,2263
+egglog/type_constraint_solver.py,sha256=MW22gwryNOqsYaB__dCCjVvdnB1Km5fovoW2IgUXaVo,5836
+egglog/visualizer.css,sha256=GVlEStdXjr1VljSajLHE59Ye1gvplHIFzCQknnVL3T0,257654
+egglog/visualizer.js,sha256=6ohS5ZbaZLYpQ42Fu7DTSHsXoCRmgwtQe5lk0PdxPhY,3788251
+egglog/visualizer_widget.py,sha256=6oFfG-mHEeeh3dzng4hjU30dNI86bKEnSXismmlJTwc,1362
+egglog/__init__.py,sha256=iUrVe5fb0XFyMCS3CwTjwhKEtU8KpIkdTpJpnUhm8o0,247
+egglog/bindings.cp310-win_amd64.pyd,sha256=9w_inf-8h95GDxA4uCRLFfxk5B8X3xi-g-nVGHsgGZs,4966400
+egglog-8.0.0.dist-info/RECORD,,

{egglog-7.1.0.dist-info → egglog-8.0.0.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: cp310-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.1.0.dist-info/RECORD

@@ -1,39 +0,0 @@
-egglog-7.1.0.dist-info/METADATA,sha256=t8YGDTCeCIpcJMDrAiicXrJAARElS-vM1R6-GY4wNzE,3829
-egglog-7.1.0.dist-info/WHEEL,sha256=keLBtIOE7ZgLyd8Cijw37iqv72a2jhfxMdDu8Unr7zo,96
-egglog-7.1.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=LFClid1ojrQFF7ezJSanwZA89j01fjJ-NkvH0VUH0Wk,16199
-egglog/egraph.py,sha256=WRu_omk-5i53wokpkSTw9wA2EqW_WI2-ZX8340_w3q8,66198
-egglog/egraph_state.py,sha256=ic6MEL41-DjozBCe-GSKqHM2vkabuC865odeg_zT964,20074
-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/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=diQNK-Uxxt6Qr58zJuQDM_BF6eKgwa-D5eYB5VJoT70,41412
-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=4YzItMaXSQw2xzId9pnjUikOojwE6G7Hg0_M4sOGJ84,19071
-egglog/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-egglog/runtime.py,sha256=P5wKThFglf2l5AynxgVAryyNjvLtFMbou7JCSSsORno,23013
-egglog/thunk.py,sha256=jYMb4owraT-3zGQEYkqMK76hk39Qdpc6bfEa7CVkjcg,1894
-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=iUrVe5fb0XFyMCS3CwTjwhKEtU8KpIkdTpJpnUhm8o0,247
-egglog/bindings.cp310-win_amd64.pyd,sha256=pjYXlaSXtC50TWNdnGwLjKbntZ0WqfM5AR9hqehz2xc,4591104
-egglog-7.1.0.dist-info/RECORD,,