tensor-network-visualization 1.6.3__tar.gz → 2.0.0__tar.gz

{tensor_network_visualization-1.6.3/src/tensor_network_visualization.egg-info → tensor_network_visualization-2.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.3
+Version: 2.0.0
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -10,18 +10,22 @@ Project-URL: Changelog, https://github.com/DOKOS-TAYOS/Tensor-Network-Visualizat
 Project-URL: Repository, https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization
 Project-URL: Issues, https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/issues
 Keywords: tensor-network,visualization,pytorch,numpy,tensorkrowch,tensornetwork,quimb,tenpy,einsum,quantum-inspired,matplotlib
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Visualization
+Classifier: Typing :: Typed
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: matplotlib>=3.7
 Requires-Dist: networkx>=3.0
-Requires-Dist: numpy
+Requires-Dist: numpy>=2.0
 Provides-Extra: dev
 Requires-Dist: build; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
@@ -49,6 +53,14 @@ Provides-Extra: einsum
 Requires-Dist: torch; extra == "einsum"
 Dynamic: license-file
 
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/tensor_network_visualization_logo.png"
+    alt="Tensor-Network-Visualization logo"
+    width="420"
+  >
+</p>
+
 # 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)
@@ -59,6 +71,47 @@ Dynamic: license-file
 Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced
 PyTorch/NumPy `einsum` tensor networks.
 
+## Gallery
+
+These are exports generated with the library and the repository demos. The clean renders use
+transparent PNGs; the controls example keeps its background so the menu stays readable.
+
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/cubic_peps_3d.png"
+    alt="Cubic PEPS 3D tensor network visualization"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/mera_2d.png"
+    alt="MERA 2D tensor network visualization with labels"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tubular_grid_3d.png"
+    alt="Tubular 3D tensor network visualization"
+    width="230"
+  >
+</p>
+
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/network_controls.png"
+    alt="MPS with controls, tensor labels, and index labels"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tensor_elements_phase.png"
+    alt="Tensor elements phase visualization"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tensor_elements_controls.png"
+    alt="Tensor elements view with the full control menus"
+    width="230"
+  >
+</p>
+
 ## Why This Exists
 
 Tensor-network libraries expose different Python objects. This package gives them a small shared

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/README.md

@@ -1,3 +1,11 @@
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/tensor_network_visualization_logo.png"
+    alt="Tensor-Network-Visualization logo"
+    width="420"
+  >
+</p>
+
 # 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)
@@ -8,6 +16,47 @@
 Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced
 PyTorch/NumPy `einsum` tensor networks.
 
+## Gallery
+
+These are exports generated with the library and the repository demos. The clean renders use
+transparent PNGs; the controls example keeps its background so the menu stays readable.
+
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/cubic_peps_3d.png"
+    alt="Cubic PEPS 3D tensor network visualization"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/mera_2d.png"
+    alt="MERA 2D tensor network visualization with labels"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tubular_grid_3d.png"
+    alt="Tubular 3D tensor network visualization"
+    width="230"
+  >
+</p>
+
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/network_controls.png"
+    alt="MPS with controls, tensor labels, and index labels"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tensor_elements_phase.png"
+    alt="Tensor elements phase visualization"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tensor_elements_controls.png"
+    alt="Tensor elements view with the full control menus"
+    width="230"
+  >
+</p>
+
 ## Why This Exists
 
 Tensor-network libraries expose different Python objects. This package gives them a small shared

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensor-network-visualization"
-version = "1.6.3"
+version = "2.0.0"
 description = "Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks."
 authors = [{ name = "Alejandro Mata Ali" }]
 readme = "README.md"
@@ -12,19 +12,23 @@ license = "MIT"
 license-files = ["LICENSE"]
 requires-python = ">=3.11"
 keywords = ["tensor-network", "visualization", "pytorch", "numpy", "tensorkrowch", "tensornetwork", "quimb", "tenpy", "einsum", "quantum-inspired", "matplotlib"]
-dependencies = [
-    "matplotlib>=3.7",
-    "networkx>=3.0",
-    "numpy",
-]
-classifiers = [
-    "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3 :: Only",
-    "Programming Language :: Python :: 3.11",
-    "Programming Language :: Python :: 3.12",
-    "Programming Language :: Python :: 3.13",
-    "Topic :: Scientific/Engineering :: Visualization",
-]
+dependencies = [
+    "matplotlib>=3.7",
+    "networkx>=3.0",
+    "numpy>=2.0",
+]
+classifiers = [
+    "Development Status :: 5 - Production/Stable",
+    "Intended Audience :: Science/Research",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Scientific/Engineering :: Visualization",
+    "Typing :: Typed",
+]
 
 [project.urls]
 Homepage = "https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization"

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0/src/tensor_network_visualization.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.3
+Version: 2.0.0
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -10,18 +10,22 @@ Project-URL: Changelog, https://github.com/DOKOS-TAYOS/Tensor-Network-Visualizat
 Project-URL: Repository, https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization
 Project-URL: Issues, https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/issues
 Keywords: tensor-network,visualization,pytorch,numpy,tensorkrowch,tensornetwork,quimb,tenpy,einsum,quantum-inspired,matplotlib
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Visualization
+Classifier: Typing :: Typed
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: matplotlib>=3.7
 Requires-Dist: networkx>=3.0
-Requires-Dist: numpy
+Requires-Dist: numpy>=2.0
 Provides-Extra: dev
 Requires-Dist: build; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
@@ -49,6 +53,14 @@ Provides-Extra: einsum
 Requires-Dist: torch; extra == "einsum"
 Dynamic: license-file
 
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/tensor_network_visualization_logo.png"
+    alt="Tensor-Network-Visualization logo"
+    width="420"
+  >
+</p>
+
 # 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)
@@ -59,6 +71,47 @@ Dynamic: license-file
 Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced
 PyTorch/NumPy `einsum` tensor networks.
 
+## Gallery
+
+These are exports generated with the library and the repository demos. The clean renders use
+transparent PNGs; the controls example keeps its background so the menu stays readable.
+
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/cubic_peps_3d.png"
+    alt="Cubic PEPS 3D tensor network visualization"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/mera_2d.png"
+    alt="MERA 2D tensor network visualization with labels"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tubular_grid_3d.png"
+    alt="Tubular 3D tensor network visualization"
+    width="230"
+  >
+</p>
+
+<p align="center">
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/network_controls.png"
+    alt="MPS with controls, tensor labels, and index labels"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tensor_elements_phase.png"
+    alt="Tensor elements phase visualization"
+    width="230"
+  >
+  <img
+    src="https://raw.githubusercontent.com/DOKOS-TAYOS/Tensor-Network-Visualization/main/images/gallery/tensor_elements_controls.png"
+    alt="Tensor elements view with the full control menus"
+    width="230"
+  >
+</p>
+
 ## Why This Exists
 
 Tensor-network libraries expose different Python objects. This package gives them a small shared

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/src/tensor_network_visualization.egg-info/SOURCES.txt

@@ -128,6 +128,7 @@ tests/test_contraction_scheme.py
 tests/test_contraction_scheme_api.py
 tests/test_contraction_viewer.py
 tests/test_demo_cli_scheme.py
+tests/test_documentation_gallery.py
 tests/test_draw_performance.py
 tests/test_edge_index_label_2d.py
 tests/test_edge_index_label_3d.py

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/src/tensor_network_visualization.egg-info/requires.txt

@@ -1,6 +1,6 @@
 matplotlib>=3.7
 networkx>=3.0
-numpy
+numpy>=2.0
 
 [dev]
 build

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/src/tensor_network_viz/_core/draw/fonts_and_scale.py

@@ -3,11 +3,13 @@ from __future__ import annotations
 import functools
 import math
 from dataclasses import dataclass
+from typing import Any, cast
 
 import numpy as np
 from matplotlib.axes import Axes
 from matplotlib.figure import Figure
 from matplotlib.font_manager import FontProperties
+from matplotlib.text import Text
 from matplotlib.textpath import TextPath
 
 from ..._matplotlib_state import (
@@ -174,33 +176,118 @@ def _draw_scale_params(
 _ZOOM_FONT_CLAMP: tuple[float, float] = (0.28, 5.5)
 
 
+def _current_2d_span(ax: Axes) -> float:
+    x0, x1 = ax.get_xlim()
+    y0, y1 = ax.get_ylim()
+    return max(float(x1 - x0), float(y1 - y0), 1e-9)
+
+
+def _zoom_font_factor(*, ref_span: float, current_span: float) -> float:
+    lo, hi = _ZOOM_FONT_CLAMP
+    factor = float(ref_span) / max(float(current_span), 1e-9)
+    return float(np.clip(factor, lo, hi))
+
+
+def _text_font_sizes(ax: Axes) -> dict[Text, float]:
+    return {text: float(text.get_fontsize()) for text in ax.texts if isinstance(text, Text)}
+
+
+def _pruned_zoom_font_sizes(
+    sizes: dict[Any, float],
+    *,
+    current_texts: tuple[Text, ...],
+) -> dict[Text, float]:
+    current_ids = {id(text) for text in current_texts}
+    return {
+        text: float(base_fontsize)
+        for text, base_fontsize in sizes.items()
+        if isinstance(text, Text) and id(text) in current_ids and text.figure is not None
+    }
+
+
+def _resolve_zoom_font_state(
+    ax: Axes,
+    *,
+    preserve_ref_span: bool,
+) -> tuple[float, dict[Text, float]]:
+    current_span = _current_2d_span(ax)
+    current_texts = tuple(text for text in ax.texts if isinstance(text, Text))
+    current_sizes = _text_font_sizes(ax)
+    if not preserve_ref_span:
+        return current_span, current_sizes
+
+    existing_state = get_zoom_font_state(ax)
+    if existing_state is None:
+        return current_span, current_sizes
+
+    ref_span = float(existing_state["ref_span"])
+    existing_sizes = _pruned_zoom_font_sizes(
+        cast(dict[Any, float], existing_state["sizes"]),
+        current_texts=current_texts,
+    )
+    resolved_sizes: dict[Text, float] = {}
+    for text in current_texts:
+        base_fontsize = existing_sizes.get(text)
+        if base_fontsize is None:
+            base_fontsize = float(current_sizes[text])
+        resolved_sizes[text] = float(base_fontsize)
+    return ref_span, resolved_sizes
+
+
+def _zoom_callback_registered(ax: Axes, signal_name: str, cid: object) -> bool:
+    callbacks = getattr(ax.callbacks, "callbacks", None)
+    if not isinstance(callbacks, dict):
+        return False
+    signal_callbacks = callbacks.get(signal_name)
+    return isinstance(signal_callbacks, dict) and cid in signal_callbacks
+
+
+def _zoom_callbacks_connected(ax: Axes, cids: list[Any]) -> bool:
+    if len(cids) != 2:
+        return False
+    return _zoom_callback_registered(ax, "xlim_changed", cids[0]) and _zoom_callback_registered(
+        ax,
+        "ylim_changed",
+        cids[1],
+    )
+
+
+def _disconnect_zoom_callbacks(ax: Axes, cids: list[Any]) -> None:
+    for cid in cids:
+        try:
+            ax.callbacks.disconnect(cid)
+        except (AttributeError, KeyError, TypeError, ValueError):
+            continue
+
+
 def _on_2d_limits_changed(ax: Axes) -> None:
     state = get_zoom_font_state(ax)
     if state is None:
         return
-    x0, x1 = ax.get_xlim()
-    y0, y1 = ax.get_ylim()
-    span = max(float(x1 - x0), float(y1 - y0), 1e-9)
-    factor = float(state["ref_span"] / span)
-    lo, hi = _ZOOM_FONT_CLAMP
-    factor = float(np.clip(factor, lo, hi))
+    factor = _zoom_font_factor(
+        ref_span=float(state["ref_span"]),
+        current_span=_current_2d_span(ax),
+    )
     for text, base_fs in state["sizes"].items():
         if text.figure is None:
             continue
         text.set_fontsize(max(3.0, base_fs * factor))
 
 
-def _register_2d_zoom_font_scaling(ax: Axes) -> None:
-    x0, x1 = ax.get_xlim()
-    y0, y1 = ax.get_ylim()
-    ref_span = max(float(x1 - x0), float(y1 - y0), 1e-9)
-    sizes = {t: float(t.get_fontsize()) for t in ax.texts}
+def _register_2d_zoom_font_scaling(
+    ax: Axes,
+    *,
+    preserve_ref_span: bool = False,
+) -> None:
+    ref_span, sizes = _resolve_zoom_font_state(ax, preserve_ref_span=preserve_ref_span)
     set_zoom_font_state(ax, ref_span=ref_span, sizes=sizes)
 
     old_cids = get_zoom_cids(ax)
-    if old_cids:
+    if _zoom_callbacks_connected(ax, old_cids):
         _on_2d_limits_changed(ax)
         return
+    if old_cids:
+        _disconnect_zoom_callbacks(ax, old_cids)
 
     def _cb(_: object) -> None:
         _on_2d_limits_changed(ax)
@@ -208,6 +295,7 @@ def _register_2d_zoom_font_scaling(ax: Axes) -> None:
     cx = ax.callbacks.connect("xlim_changed", _cb)
     cy = ax.callbacks.connect("ylim_changed", _cb)
     set_zoom_cids(ax, [cx, cy])
+    _on_2d_limits_changed(ax)
 
 
 __all__ = [

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/src/tensor_network_viz/_core/draw/graph_pipeline.py

@@ -236,7 +236,7 @@ def _draw_graph(
                 include_viewer=False,
             )
     if build_scene_state:
-        from ..._matplotlib_state import set_contraction_controls, set_scene
+        from ..._matplotlib_state import set_contraction_controls, set_home_view, set_scene
         from .render_prep import _build_interactive_scene_state
 
         scene = _build_interactive_scene_state(
@@ -250,6 +250,7 @@ def _draw_graph(
         )
         scene.contraction_controls = controls
         set_scene(ax, scene)
+        set_home_view(ax, ax._get_view())
         if controls is not None:
             set_contraction_controls(ax, controls)
             controls.bind_scene(scene)

{tensor_network_visualization-1.6.3 → tensor_network_visualization-2.0.0}/src/tensor_network_viz/_core/layout/direction_common.py

@@ -210,7 +210,11 @@ def _component_centroid(
     *,
     dimensions: int,
 ) -> np.ndarray:
-    source_ids = component.anchor_node_ids or component.visible_node_ids or component.node_ids
+    source_ids = (
+        component.anchor_node_ids
+        or component.geometry_visible_node_ids
+        or component.geometry_node_ids
+    )
     anchor_ids = tuple(node_id for node_id in source_ids if node_id in positions)
     if not anchor_ids:
         anchor_ids = tuple(sorted(positions))
@@ -227,7 +231,11 @@ def _component_main_axis_2d_like(
     component: _LayoutComponent,
     positions: NodePositions,
 ) -> np.ndarray:
-    source_ids = component.anchor_node_ids or component.visible_node_ids or component.node_ids
+    source_ids = (
+        component.anchor_node_ids
+        or component.geometry_visible_node_ids
+        or component.geometry_node_ids
+    )
     anchor_ids = tuple(node_id for node_id in source_ids if node_id in positions)
     if not anchor_ids:
         anchor_ids = tuple(sorted(positions))