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

{tensor_network_visualization-1.6.2 → tensor_network_visualization-1.6.3}/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-1.6.2/src/tensor_network_visualization.egg-info → tensor_network_visualization-1.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.2
+Version: 1.6.3
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.2
+Version: 1.6.3
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT

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

@@ -7,9 +7,11 @@ src/tensor_network_visualization.egg-info/dependency_links.txt
 src/tensor_network_visualization.egg-info/requires.txt
 src/tensor_network_visualization.egg-info/top_level.txt
 src/tensor_network_viz/__init__.py
+src/tensor_network_viz/_auto_display.py
 src/tensor_network_viz/_contraction_viewer_style.py
 src/tensor_network_viz/_contraction_viewer_ui.py
 src/tensor_network_viz/_engine_specs.py
+src/tensor_network_viz/_import_state.py
 src/tensor_network_viz/_input_inspection.py
 src/tensor_network_viz/_interactive_scene.py
 src/tensor_network_viz/_logging.py
@@ -118,6 +120,7 @@ src/tensor_network_viz/tensorkrowch/renderer.py
 src/tensor_network_viz/tensornetwork/__init__.py
 src/tensor_network_viz/tensornetwork/graph.py
 src/tensor_network_viz/tensornetwork/renderer.py
+tests/test_bench_user_workflows.py
 tests/test_ci_workflow.py
 tests/test_contraction_cost.py
 tests/test_contraction_groups_once.py

tensor_network_visualization-1.6.3/src/tensor_network_viz/_auto_display.py

@@ -0,0 +1,54 @@
+"""Shared helpers for auto-resolving display toggles."""
+
+from __future__ import annotations
+
+from dataclasses import replace
+
+from ._core.graph import _GraphData
+from .config import PlotConfig
+
+AUTO_VISIBLE_TENSOR_THRESHOLD: int = 25
+
+
+def count_visible_tensors(graph: _GraphData) -> int:
+    """Count visible non-virtual tensors, falling back to total nodes when needed."""
+    visible_tensors = sum(1 for node in graph.nodes.values() if not node.is_virtual)
+    return visible_tensors or len(graph.nodes)
+
+
+def resolve_auto_display_flag(
+    value: bool | None,
+    *,
+    visible_tensor_count: int,
+) -> bool:
+    """Resolve one optional display flag using the shared visibility heuristic."""
+    if value is not None:
+        return bool(value)
+    return int(visible_tensor_count) < AUTO_VISIBLE_TENSOR_THRESHOLD
+
+
+def resolve_auto_display_config(
+    config: PlotConfig,
+    *,
+    visible_tensor_count: int,
+) -> PlotConfig:
+    """Freeze auto display flags into concrete booleans for the current render."""
+    return replace(
+        config,
+        show_nodes=resolve_auto_display_flag(
+            config.show_nodes,
+            visible_tensor_count=visible_tensor_count,
+        ),
+        show_tensor_labels=resolve_auto_display_flag(
+            config.show_tensor_labels,
+            visible_tensor_count=visible_tensor_count,
+        ),
+    )
+
+
+__all__ = [
+    "AUTO_VISIBLE_TENSOR_THRESHOLD",
+    "count_visible_tensors",
+    "resolve_auto_display_config",
+    "resolve_auto_display_flag",
+]

{tensor_network_visualization-1.6.2 → tensor_network_visualization-1.6.3}/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-1.6.2 → tensor_network_visualization-1.6.3}/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",
+]