tensor-network-visualization 1.6.0__tar.gz → 1.6.1__tar.gz

{tensor_network_visualization-1.6.0/src/tensor_network_visualization.egg-info → tensor_network_visualization-1.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.0
+Version: 1.6.1
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -169,11 +169,38 @@ figures with embedded controls. See [Installation](docs/installation.md) and
   logging.
 - [User Guide](docs/guide.md): workflows, notebooks, exports, layouts, tensor inspection,
   comparisons, snapshots, and performance tips.
+- [Layout Algorithms](docs/algorithms.md): node placement and free-edge direction rules in 2D and
+  3D.
 - [Backend Examples](docs/backends.md): copy-paste examples for TensorKrowch, TensorNetwork, Quimb,
   TeNPy, and `einsum`.
 - [Troubleshooting](docs/troubleshooting.md): common install, Jupyter, Matplotlib, backend, and data
   issues.
 - [Repository Examples](examples/README.md): command-line demo launcher and example catalog.
+- [Demo Commands](commands.md): copy-paste commands for every repository demo.
+
+## Demo Gallery
+
+The repository examples are organized around the same launcher:
+
+```bash
+python examples/run_demo.py <group> <demo>
+```
+
+The gallery includes backend demos for TensorKrowch, TensorNetwork, Quimb, TeNPy, and `einsum`,
+plus three focused groups:
+
+- `themes overview`: compares `default`, `paper`, and `colorblind` visual modes.
+- `placements`: shows object, list, 2D grid, 3D grid, manual positions, manual schemes, and named
+  index inputs.
+- `geometry`: renders larger irregular, incomplete, triangular, pyramidal, circular, and
+  disconnected networks.
+
+For batch checks, use:
+
+```bash
+python examples/run_all_examples.py --group engines --views 2d --list
+python examples/run_all_examples.py --group all --views 2d --output-dir .tmp/examples
+```
 
 ## Project Links
 

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1}/README.md

@@ -1,133 +1,160 @@
-# Tensor-Network-Visualization
-
-[![CI](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/actions/workflows/ci.yml/badge.svg)](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/actions/workflows/ci.yml)
-[![PyPI version](https://img.shields.io/pypi/v/tensor-network-visualization.svg)](https://pypi.org/project/tensor-network-visualization/)
-[![Python versions](https://img.shields.io/pypi/pyversions/tensor-network-visualization.svg)](https://pypi.org/project/tensor-network-visualization/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-
-Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced
-PyTorch/NumPy `einsum` tensor networks.
-
-## Why This Exists
-
-Tensor-network libraries expose different Python objects. This package gives them a small shared
-visualization API so you can inspect structure, tensor values, contraction playback, and normalized
-graph exports without rewriting plotting code for every backend.
-
-The common entry points are:
-
-```python
-show_tensor_network(...)
-show_tensor_elements(...)
-show_tensor_comparison(...)
-normalize_tensor_network(...)
-export_tensor_network_snapshot(...)
-```
-
-## Install
-
-- PyPI package name: `tensor-network-visualization`
-- Import package: `tensor_network_viz`
-- Requires Python 3.11 or newer.
-
-```bash
-python -m pip install tensor-network-visualization
-```
-
-The base install only depends on `numpy`, `matplotlib`, and `networkx`.
-
-For interactive Jupyter figures:
-
-```bash
-python -m pip install "tensor-network-visualization[jupyter]"
-```
-
-For backend-specific packages, install the matching extra, for example:
-
-```bash
-python -m pip install "tensor-network-visualization[quimb]"
-```
-
-See [docs/installation.md](docs/installation.md) for virtual environments, all optional extras, and
-local development installs.
-
-## Basic Usage
-
-### NumPy `einsum` trace (base install)
-
-This example uses only base dependencies and a NumPy-backed `EinsumTrace`.
-
-```python
-import numpy as np
-from tensor_network_viz import EinsumTrace, PlotConfig, einsum, show_tensor_network
-
-trace = EinsumTrace()
-a = np.ones((2, 3), dtype=float)
-x = np.array([1.0, -0.5, 0.25], dtype=float)
-
-trace.bind("A", a)
-trace.bind("x", x)
-einsum("ab,b->a", a, x, trace=trace, backend="numpy")
-
-fig, ax = show_tensor_network(
-    trace,
-    config=PlotConfig(show_tensor_labels=True, hover_labels=True),
-    show=False,
-)
-fig.savefig("einsum-network.png", bbox_inches="tight")
-```
-
-### TensorKrowch
-
-Install the TensorKrowch extra (see [Installation](docs/installation.md) for details):
-
-```bash
-python -m pip install "tensor-network-visualization[tensorkrowch]"
-```
-
-```python
-import tensorkrowch as tk
-from tensor_network_viz import PlotConfig, show_tensor_network
-
-network = tk.TensorNetwork(name="demo")
-left = tk.Node(shape=(2, 2), axes_names=("a", "b"), name="L", network=network)
-right = tk.Node(shape=(2, 2), axes_names=("b", "c"), name="R", network=network)
-left["b"] ^ right["b"]
-
-fig, ax = show_tensor_network(
-    network,
-    config=PlotConfig(show_tensor_labels=True, show_index_labels=False),
-    show=False,
-)
-fig.savefig("tensorkrowch-network.png", bbox_inches="tight")
-```
-
-Use `show=False` when you want to save or customize the figure yourself. Use
-`show_controls=False` when you want a clean static figure with no embedded Matplotlib controls.
-
-In Jupyter, install `"tensor-network-visualization[jupyter]"`, run `%matplotlib widget` before
-plotting, then call `show_tensor_network` as usual (default `show_controls=True`) for interactive
-figures with embedded controls. See [Installation](docs/installation.md) and
-[User Guide](docs/guide.md) for details.
-
-## Documentation
-
-- [Installation](docs/installation.md): virtual environments, optional extras, Jupyter, and local
-  editable installs.
-- [API Reference](docs/api.md): public functions, configuration objects, snapshots, exceptions, and
-  logging.
-- [User Guide](docs/guide.md): workflows, notebooks, exports, layouts, tensor inspection,
-  comparisons, snapshots, and performance tips.
-- [Backend Examples](docs/backends.md): copy-paste examples for TensorKrowch, TensorNetwork, Quimb,
-  TeNPy, and `einsum`.
-- [Troubleshooting](docs/troubleshooting.md): common install, Jupyter, Matplotlib, backend, and data
-  issues.
-- [Repository Examples](examples/README.md): command-line demo launcher and example catalog.
-
-## Project Links
-
-- [Changelog](CHANGELOG.md)
-- [Contributing](CONTRIBUTING.md)
-- [Code of Conduct](CODE_OF_CONDUCT.md)
-- [Issues](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/issues)
-- [Repository](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization)
+# Tensor-Network-Visualization
+
+[![CI](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/actions/workflows/ci.yml/badge.svg)](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/actions/workflows/ci.yml)
+[![PyPI version](https://img.shields.io/pypi/v/tensor-network-visualization.svg)](https://pypi.org/project/tensor-network-visualization/)
+[![Python versions](https://img.shields.io/pypi/pyversions/tensor-network-visualization.svg)](https://pypi.org/project/tensor-network-visualization/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced
+PyTorch/NumPy `einsum` tensor networks.
+
+## Why This Exists
+
+Tensor-network libraries expose different Python objects. This package gives them a small shared
+visualization API so you can inspect structure, tensor values, contraction playback, and normalized
+graph exports without rewriting plotting code for every backend.
+
+The common entry points are:
+
+```python
+show_tensor_network(...)
+show_tensor_elements(...)
+show_tensor_comparison(...)
+normalize_tensor_network(...)
+export_tensor_network_snapshot(...)
+```
+
+## Install
+
+- PyPI package name: `tensor-network-visualization`
+- Import package: `tensor_network_viz`
+- Requires Python 3.11 or newer.
+
+```bash
+python -m pip install tensor-network-visualization
+```
+
+The base install only depends on `numpy`, `matplotlib`, and `networkx`.
+
+For interactive Jupyter figures:
+
+```bash
+python -m pip install "tensor-network-visualization[jupyter]"
+```
+
+For backend-specific packages, install the matching extra, for example:
+
+```bash
+python -m pip install "tensor-network-visualization[quimb]"
+```
+
+See [docs/installation.md](docs/installation.md) for virtual environments, all optional extras, and
+local development installs.
+
+## Basic Usage
+
+### NumPy `einsum` trace (base install)
+
+This example uses only base dependencies and a NumPy-backed `EinsumTrace`.
+
+```python
+import numpy as np
+from tensor_network_viz import EinsumTrace, PlotConfig, einsum, show_tensor_network
+
+trace = EinsumTrace()
+a = np.ones((2, 3), dtype=float)
+x = np.array([1.0, -0.5, 0.25], dtype=float)
+
+trace.bind("A", a)
+trace.bind("x", x)
+einsum("ab,b->a", a, x, trace=trace, backend="numpy")
+
+fig, ax = show_tensor_network(
+    trace,
+    config=PlotConfig(show_tensor_labels=True, hover_labels=True),
+    show=False,
+)
+fig.savefig("einsum-network.png", bbox_inches="tight")
+```
+
+### TensorKrowch
+
+Install the TensorKrowch extra (see [Installation](docs/installation.md) for details):
+
+```bash
+python -m pip install "tensor-network-visualization[tensorkrowch]"
+```
+
+```python
+import tensorkrowch as tk
+from tensor_network_viz import PlotConfig, show_tensor_network
+
+network = tk.TensorNetwork(name="demo")
+left = tk.Node(shape=(2, 2), axes_names=("a", "b"), name="L", network=network)
+right = tk.Node(shape=(2, 2), axes_names=("b", "c"), name="R", network=network)
+left["b"] ^ right["b"]
+
+fig, ax = show_tensor_network(
+    network,
+    config=PlotConfig(show_tensor_labels=True, show_index_labels=False),
+    show=False,
+)
+fig.savefig("tensorkrowch-network.png", bbox_inches="tight")
+```
+
+Use `show=False` when you want to save or customize the figure yourself. Use
+`show_controls=False` when you want a clean static figure with no embedded Matplotlib controls.
+
+In Jupyter, install `"tensor-network-visualization[jupyter]"`, run `%matplotlib widget` before
+plotting, then call `show_tensor_network` as usual (default `show_controls=True`) for interactive
+figures with embedded controls. See [Installation](docs/installation.md) and
+[User Guide](docs/guide.md) for details.
+
+## Documentation
+
+- [Installation](docs/installation.md): virtual environments, optional extras, Jupyter, and local
+  editable installs.
+- [API Reference](docs/api.md): public functions, configuration objects, snapshots, exceptions, and
+  logging.
+- [User Guide](docs/guide.md): workflows, notebooks, exports, layouts, tensor inspection,
+  comparisons, snapshots, and performance tips.
+- [Layout Algorithms](docs/algorithms.md): node placement and free-edge direction rules in 2D and
+  3D.
+- [Backend Examples](docs/backends.md): copy-paste examples for TensorKrowch, TensorNetwork, Quimb,
+  TeNPy, and `einsum`.
+- [Troubleshooting](docs/troubleshooting.md): common install, Jupyter, Matplotlib, backend, and data
+  issues.
+- [Repository Examples](examples/README.md): command-line demo launcher and example catalog.
+- [Demo Commands](commands.md): copy-paste commands for every repository demo.
+
+## Demo Gallery
+
+The repository examples are organized around the same launcher:
+
+```bash
+python examples/run_demo.py <group> <demo>
+```
+
+The gallery includes backend demos for TensorKrowch, TensorNetwork, Quimb, TeNPy, and `einsum`,
+plus three focused groups:
+
+- `themes overview`: compares `default`, `paper`, and `colorblind` visual modes.
+- `placements`: shows object, list, 2D grid, 3D grid, manual positions, manual schemes, and named
+  index inputs.
+- `geometry`: renders larger irregular, incomplete, triangular, pyramidal, circular, and
+  disconnected networks.
+
+For batch checks, use:
+
+```bash
+python examples/run_all_examples.py --group engines --views 2d --list
+python examples/run_all_examples.py --group all --views 2d --output-dir .tmp/examples
+```
+
+## Project Links
+
+- [Changelog](CHANGELOG.md)
+- [Contributing](CONTRIBUTING.md)
+- [Code of Conduct](CODE_OF_CONDUCT.md)
+- [Issues](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/issues)
+- [Repository](https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization)

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensor-network-visualization"
-version = "1.6.0"
+version = "1.6.1"
 description = "Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks."
 authors = [{ name = "Alejandro Mata Ali" }]
 readme = "README.md"

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1/src/tensor_network_visualization.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.0
+Version: 1.6.1
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -169,11 +169,38 @@ figures with embedded controls. See [Installation](docs/installation.md) and
   logging.
 - [User Guide](docs/guide.md): workflows, notebooks, exports, layouts, tensor inspection,
   comparisons, snapshots, and performance tips.
+- [Layout Algorithms](docs/algorithms.md): node placement and free-edge direction rules in 2D and
+  3D.
 - [Backend Examples](docs/backends.md): copy-paste examples for TensorKrowch, TensorNetwork, Quimb,
   TeNPy, and `einsum`.
 - [Troubleshooting](docs/troubleshooting.md): common install, Jupyter, Matplotlib, backend, and data
   issues.
 - [Repository Examples](examples/README.md): command-line demo launcher and example catalog.
+- [Demo Commands](commands.md): copy-paste commands for every repository demo.
+
+## Demo Gallery
+
+The repository examples are organized around the same launcher:
+
+```bash
+python examples/run_demo.py <group> <demo>
+```
+
+The gallery includes backend demos for TensorKrowch, TensorNetwork, Quimb, TeNPy, and `einsum`,
+plus three focused groups:
+
+- `themes overview`: compares `default`, `paper`, and `colorblind` visual modes.
+- `placements`: shows object, list, 2D grid, 3D grid, manual positions, manual schemes, and named
+  index inputs.
+- `geometry`: renders larger irregular, incomplete, triangular, pyramidal, circular, and
+  disconnected networks.
+
+For batch checks, use:
+
+```bash
+python examples/run_all_examples.py --group engines --views 2d --list
+python examples/run_all_examples.py --group all --views 2d --output-dir .tmp/examples
+```
 
 ## Project Links
 

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1}/src/tensor_network_visualization.egg-info/SOURCES.txt

@@ -75,6 +75,7 @@ src/tensor_network_viz/_core/layout/direction_common.py
 src/tensor_network_viz/_core/layout/force_directed.py
 src/tensor_network_viz/_core/layout/free_directions_2d.py
 src/tensor_network_viz/_core/layout/free_directions_3d.py
+src/tensor_network_viz/_core/layout/generic_coarsening.py
 src/tensor_network_viz/_core/layout/geometry.py
 src/tensor_network_viz/_core/layout/parameters.py
 src/tensor_network_viz/_core/layout/positions.py

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1}/src/tensor_network_viz/__init__.py

@@ -16,6 +16,7 @@ if TYPE_CHECKING:
     from .config import (
         EngineName,
         PlotConfig,
+        PlotTheme,
         TensorNetworkDiagnosticsConfig,
         TensorNetworkFocus,
         ViewName,
@@ -161,6 +162,7 @@ _LAZY_EXPORTS: dict[str, tuple[str, str]] = {
     "NormalizedTensorGraph": (".snapshot", "NormalizedTensorGraph"),
     "NormalizedTensorNode": (".snapshot", "NormalizedTensorNode"),
     "PlotConfig": (".config", "PlotConfig"),
+    "PlotTheme": (".config", "PlotTheme"),
     "TenPyTensorNetwork": (".tenpy.explicit", "TenPyTensorNetwork"),
     "TensorComparisonConfig": (".tensor_comparison_config", "TensorComparisonConfig"),
     "TensorAnalysisConfig": (".tensor_elements_config", "TensorAnalysisConfig"),
@@ -214,6 +216,7 @@ __all__ = [
     "NormalizedTensorGraph",
     "NormalizedTensorNode",
     "PlotConfig",
+    "PlotTheme",
     "TensorComparisonConfig",
     "TensorAnalysisConfig",
     "TensorDataError",

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1}/src/tensor_network_viz/_contraction_viewer_ui.py

@@ -8,7 +8,9 @@ from matplotlib.figure import Figure
 from matplotlib.text import Text
 from matplotlib.widgets import Button, Slider
 
+from ._ui_utils import _control_slider_handle_style, _slider_track_color, _style_control_button
 from ._widgets import _SafeButton, _SafeSlider
+from .config import PlotConfig
 
 _PLAYBACK_DETAILS_BOUNDS: tuple[float, float, float, float] = (0.23, 0.116, 0.7, 0.12)
 # Top of the cost / step-details axis; interactive chrome (checkboxes, 2d/3d) aligns to this y.
@@ -29,6 +31,7 @@ _PLAYBACK_SLIDER_HANDLE_STYLE: dict[str, Any] = {
     "edgecolor": "#1d4ed8",
     "size": 11,
 }
+_PLAYBACK_SLIDER_TRACK_WHITE_MIX: float = 0.82
 _PLAYBACK_BUTTON_START_X: float = 0.73
 _PLAYBACK_BUTTON_Y: float = 0.058
 _PLAYBACK_BUTTON_WIDTH: float = 0.055
@@ -51,6 +54,15 @@ _CONTROL_FRAME_PROPS: dict[str, float] = {"s": 44.0, "linewidth": 0.9}
 _CONTROL_CHECK_PROPS: dict[str, float] = {"s": 34.0, "linewidth": 1.0}
 
 
+def _playback_slider_handle_style(config: PlotConfig | None) -> dict[str, Any]:
+    resolved = config if config is not None else PlotConfig()
+    return _control_slider_handle_style(
+        active_color=resolved.bond_edge_color,
+        edge_color=resolved.node_edge_color,
+        size=float(_PLAYBACK_SLIDER_HANDLE_STYLE["size"]),
+    )
+
+
 def create_playback_details_panel(fig: Figure) -> tuple[Axes, Text]:
     ax_details = fig.add_axes(_PLAYBACK_DETAILS_BOUNDS)
     ax_details.set_xticks([])
@@ -82,8 +94,11 @@ def create_playback_slider(
     *,
     num_steps: int,
     initial_step: int,
+    config: PlotConfig | None = None,
 ) -> Slider:
     ax_slider = fig.add_axes(_PLAYBACK_SLIDER_BOUNDS)
+    resolved = config if config is not None else PlotConfig()
+    active_color = resolved.bond_edge_color
     return _SafeSlider(
         ax_slider,
         "Step",
@@ -91,7 +106,12 @@ def create_playback_slider(
         float(max(0, num_steps)),
         valinit=float(initial_step),
         valstep=1,
-        handle_style=_PLAYBACK_SLIDER_HANDLE_STYLE,
+        color=active_color,
+        track_color=_slider_track_color(
+            active_color,
+            white_mix=_PLAYBACK_SLIDER_TRACK_WHITE_MIX,
+        ),
+        handle_style=_playback_slider_handle_style(config),
     )
 
 
@@ -104,11 +124,12 @@ def create_playback_buttons(fig: Figure) -> tuple[Button, Button, Button]:
     ax_play = fig.add_axes((bx, by, bw, bh))
     ax_pause = fig.add_axes((bx + bw + gap, by, bw, bh))
     ax_reset = fig.add_axes((bx + 2.0 * (bw + gap), by, _PLAYBACK_RESET_WIDTH, bh))
-    return (
-        _SafeButton(ax_play, "Play"),
-        _SafeButton(ax_pause, "Pause"),
-        _SafeButton(ax_reset, "Reset"),
-    )
+    btn_play = _SafeButton(ax_play, "Play")
+    btn_pause = _SafeButton(ax_pause, "Pause")
+    btn_reset = _SafeButton(ax_reset, "Reset")
+    for button in (btn_play, btn_pause, btn_reset):
+        _style_control_button(button, font_size=8.8)
+    return (btn_play, btn_pause, btn_reset)
 
 
 __all__ = [

{tensor_network_visualization-1.6.0 → tensor_network_visualization-1.6.1}/src/tensor_network_viz/_core/draw/constants.py

@@ -6,7 +6,7 @@ import numpy as np
 
 # Multiedge separation; keep in sync with
 # `_LAYOUT_BOND_CURVE_OFFSET_FACTOR` in `layout/parameters.py`.
-_CURVE_OFFSET_FACTOR: float = 0.15
+_CURVE_OFFSET_FACTOR: float = 0.10
 # Blends with chord length so multiedges keep visible separation when endpoints are close.
 _CURVE_NEAR_PAIR_REF: float = 0.28
 # Extra radius + offset so index captions sit just outside tensor disks (data units).