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

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.1}/LICENSE

@@ -1,21 +1,21 @@
-MIT License
-
-Copyright (c) 2026 Alejandro Mata Ali
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+MIT License
+
+Copyright (c) 2026 Alejandro Mata Ali
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 2.0.0
+Version: 2.0.1
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -259,7 +259,9 @@ 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.
+- `themes overview`: compares the `PlotConfig` visual presets for tensor-network figures.
+- `themes tensor_elements`: compares the `TensorElementsConfig` presets for `show_tensor_elements(...)`
+  and the linked tensor inspector.
 - `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

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

@@ -1,67 +1,67 @@
-<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)
-[![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/)
+<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)
+[![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.
-
-## 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
-visualization API so you can inspect structure, tensor values, contraction playback, and normalized
-graph exports without rewriting plotting code for every backend.
+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
+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:
 
@@ -201,14 +201,16 @@ The repository examples are organized around the same launcher:
 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.
+The gallery includes backend demos for TensorKrowch, TensorNetwork, Quimb, TeNPy, and `einsum`,
+plus three focused groups:
+
+- `themes overview`: compares the `PlotConfig` visual presets for tensor-network figures.
+- `themes tensor_elements`: compares the `TensorElementsConfig` presets for `show_tensor_elements(...)`
+  and the linked tensor inspector.
+- `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:
 

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensor-network-visualization"
-version = "2.0.0"
+version = "2.0.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"
@@ -12,23 +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>=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",
-]
+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-2.0.0 → tensor_network_visualization-2.0.1/src/tensor_network_visualization.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 2.0.0
+Version: 2.0.1
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -259,7 +259,9 @@ 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.
+- `themes overview`: compares the `PlotConfig` visual presets for tensor-network figures.
+- `themes tensor_elements`: compares the `TensorElementsConfig` presets for `show_tensor_elements(...)`
+  and the linked tensor inspector.
 - `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

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.1}/src/tensor_network_viz/__init__.py

@@ -48,7 +48,11 @@ if TYPE_CHECKING:
     from .tensor_comparison import show_tensor_comparison
     from .tensor_comparison_config import TensorComparisonConfig
     from .tensor_elements import show_tensor_elements
-    from .tensor_elements_config import TensorAnalysisConfig, TensorElementsConfig
+    from .tensor_elements_config import (
+        TensorAnalysisConfig,
+        TensorElementsConfig,
+        TensorElementsTheme,
+    )
     from .viewer import show_tensor_network
 else:
 
@@ -169,6 +173,7 @@ _LAZY_EXPORTS: dict[str, tuple[str, str]] = {
     "TensorDataError": (".exceptions", "TensorDataError"),
     "TensorDataTypeError": (".exceptions", "TensorDataTypeError"),
     "TensorElementsConfig": (".tensor_elements_config", "TensorElementsConfig"),
+    "TensorElementsTheme": (".tensor_elements_config", "TensorElementsTheme"),
     "TensorNetworkDiagnosticsConfig": (".config", "TensorNetworkDiagnosticsConfig"),
     "TensorNetworkFocus": (".config", "TensorNetworkFocus"),
     "TensorNetworkLayoutSnapshot": (".snapshot", "TensorNetworkLayoutSnapshot"),
@@ -222,6 +227,7 @@ __all__ = [
     "TensorDataError",
     "TensorDataTypeError",
     "TensorElementsConfig",
+    "TensorElementsTheme",
     "TensorNetworkDiagnosticsConfig",
     "TensorNetworkFocus",
     "TensorNetworkLayoutSnapshot",

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.1}/src/tensor_network_viz/_core/__init__.py

@@ -1 +1 @@
-"""Internal shared rendering core for tensor network visualization."""
+"""Internal shared rendering core for tensor network visualization."""

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.1}/src/tensor_network_viz/_core/draw/disk_metrics.py

@@ -1,73 +1,73 @@
-from __future__ import annotations
-
-import math
-from typing import Any, Literal, cast
-
-import numpy as np
-from matplotlib.axes import Axes
-from mpl_toolkits.mplot3d import proj3d
-
-from .fonts_and_scale import _DrawScaleParams
-
-
-def _tensor_disk_radius_px_3d_nominal(ax: Any, p: _DrawScaleParams) -> float:
-    """Pixels per data unit from axis spans only (no per-point projection); ``r`` in data units."""
-    from .viewport_geometry import _nominal_figure_px_per_data_unit_3d
-
-    k = _nominal_figure_px_per_data_unit_3d(ax)
-    return float(max(float(p.r) * k, 1e-9))
-
-
-def _display_disk_radius_px_2d(ax: Axes, center: np.ndarray, r_data: float) -> float:
-    """Pixel radius for horizontal *r_data* at *center* (equal-aspect 2D)."""
-    c = np.asarray(center[:2], dtype=float)
-    row0 = c.reshape(1, -1)
-    row1 = (c + np.array([float(r_data), 0.0], dtype=float)).reshape(1, -1)
-    t0 = ax.transData.transform(row0)[0]
-    t1 = ax.transData.transform(row1)[0]
-    return float(np.hypot(float(t0[0] - t1[0]), float(t0[1] - t1[1])))
-
-
-def _display_disk_radius_px_3d(ax: Any, center: np.ndarray, r_data: float) -> float:
-    """Conservative screen radius for data-space sphere *r_data* (current 3D view)."""
-    c = np.asarray(center, dtype=float)
-    r = float(r_data)
-    M = ax.get_proj()
-    xs0, ys0, _zs0 = proj3d.proj_transform(c[0], c[1], c[2], M)
-    pt_center = ax.transData.transform((xs0, ys0))
-    md = math.inf
-    for ex, ey, ez in (
-        (1.0, 0.0, 0.0),
-        (-1.0, 0.0, 0.0),
-        (0.0, 1.0, 0.0),
-        (0.0, -1.0, 0.0),
-        (0.0, 0.0, 1.0),
-        (0.0, 0.0, -1.0),
-    ):
-        p = c + r * np.array([ex, ey, ez], dtype=float)
-        xs, ys, _zs = proj3d.proj_transform(p[0], p[1], p[2], M)
-        pt = ax.transData.transform((xs, ys))
-        d = float(np.hypot(pt[0] - pt_center[0], pt[1] - pt_center[1]))
-        md = min(md, d)
-    if not math.isfinite(md):
-        return 0.0
-    return float(md)
-
-
-def _tensor_disk_radius_px(
-    ax: Any,
-    anchor: np.ndarray,
-    p: _DrawScaleParams,
-    dimensions: Literal[2, 3],
-) -> float:
-    if dimensions == 2:
-        return _display_disk_radius_px_2d(cast(Axes, ax), anchor, p.r)
-    return _display_disk_radius_px_3d(ax, anchor, p.r)
-
-
-__all__ = [
-    "_display_disk_radius_px_2d",
-    "_display_disk_radius_px_3d",
-    "_tensor_disk_radius_px",
-    "_tensor_disk_radius_px_3d_nominal",
-]
+from __future__ import annotations
+
+import math
+from typing import Any, Literal, cast
+
+import numpy as np
+from matplotlib.axes import Axes
+from mpl_toolkits.mplot3d import proj3d
+
+from .fonts_and_scale import _DrawScaleParams
+
+
+def _tensor_disk_radius_px_3d_nominal(ax: Any, p: _DrawScaleParams) -> float:
+    """Pixels per data unit from axis spans only (no per-point projection); ``r`` in data units."""
+    from .viewport_geometry import _nominal_figure_px_per_data_unit_3d
+
+    k = _nominal_figure_px_per_data_unit_3d(ax)
+    return float(max(float(p.r) * k, 1e-9))
+
+
+def _display_disk_radius_px_2d(ax: Axes, center: np.ndarray, r_data: float) -> float:
+    """Pixel radius for horizontal *r_data* at *center* (equal-aspect 2D)."""
+    c = np.asarray(center[:2], dtype=float)
+    row0 = c.reshape(1, -1)
+    row1 = (c + np.array([float(r_data), 0.0], dtype=float)).reshape(1, -1)
+    t0 = ax.transData.transform(row0)[0]
+    t1 = ax.transData.transform(row1)[0]
+    return float(np.hypot(float(t0[0] - t1[0]), float(t0[1] - t1[1])))
+
+
+def _display_disk_radius_px_3d(ax: Any, center: np.ndarray, r_data: float) -> float:
+    """Conservative screen radius for data-space sphere *r_data* (current 3D view)."""
+    c = np.asarray(center, dtype=float)
+    r = float(r_data)
+    M = ax.get_proj()
+    xs0, ys0, _zs0 = proj3d.proj_transform(c[0], c[1], c[2], M)
+    pt_center = ax.transData.transform((xs0, ys0))
+    md = math.inf
+    for ex, ey, ez in (
+        (1.0, 0.0, 0.0),
+        (-1.0, 0.0, 0.0),
+        (0.0, 1.0, 0.0),
+        (0.0, -1.0, 0.0),
+        (0.0, 0.0, 1.0),
+        (0.0, 0.0, -1.0),
+    ):
+        p = c + r * np.array([ex, ey, ez], dtype=float)
+        xs, ys, _zs = proj3d.proj_transform(p[0], p[1], p[2], M)
+        pt = ax.transData.transform((xs, ys))
+        d = float(np.hypot(pt[0] - pt_center[0], pt[1] - pt_center[1]))
+        md = min(md, d)
+    if not math.isfinite(md):
+        return 0.0
+    return float(md)
+
+
+def _tensor_disk_radius_px(
+    ax: Any,
+    anchor: np.ndarray,
+    p: _DrawScaleParams,
+    dimensions: Literal[2, 3],
+) -> float:
+    if dimensions == 2:
+        return _display_disk_radius_px_2d(cast(Axes, ax), anchor, p.r)
+    return _display_disk_radius_px_3d(ax, anchor, p.r)
+
+
+__all__ = [
+    "_display_disk_radius_px_2d",
+    "_display_disk_radius_px_3d",
+    "_tensor_disk_radius_px",
+    "_tensor_disk_radius_px_3d_nominal",
+]