tensor-network-visualization 1.5.1__tar.gz → 1.5.2__tar.gz

{tensor_network_visualization-1.5.1/src/tensor_network_visualization.egg-info → tensor_network_visualization-1.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.5.1
+Version: 1.5.2
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -171,13 +171,14 @@ show_tensor_elements(
 
 When several tensors are present, the figure keeps one tensor active at a time and uses a slider
 to switch between them. The interactive controls are grouped: `basic` (`elements`, `magnitude`,
-`distribution`, `data`), `complex` (`real`, `imag`, `phase`), and `diagnostic` (`sign`,
-`signed_value`).
+`log_magnitude`, `distribution`, `data`), `complex` (`real`, `imag`, `phase`), and
+`diagnostic` (`sign`, `signed_value`, `sparsity`, `nan_inf`).
 
 - `data`: single tensor, iterable of tensors, supported backend-native tensor collections, or an
   `EinsumTrace` with live tensor values.
 - `engine`: optional backend override. If omitted, the library auto-detects it.
 - `config`: tensor-inspection behavior lives here.
+- `data` mode now includes global stats, per-axis summaries, and top-k coordinates by magnitude.
 - `ax`: render a single tensor into an existing Matplotlib axis.
 - `show_controls`: if `True`, add compact Matplotlib `group + mode` controls and, when needed, a tensor slider.
 - `show`: if `False`, nothing is displayed automatically.
@@ -220,6 +221,10 @@ config = TensorElementsConfig(
     mode="auto",
     max_matrix_shape=(256, 256),
     histogram_bins=40,
+    topk_count=8,
+    robust_percentiles=(1.0, 99.0),
+    shared_color_scale=False,
+    highlight_outliers=False,
 )
 ```
 
@@ -228,10 +233,13 @@ This is where you configure:
 - the active inspection mode,
 - row/column axis grouping for rank > 2 tensors,
 - heatmap downsampling limits,
-- histogram sampling and bin count.
+- histogram sampling and bin count,
+- data-summary depth (`topk_count`),
+- optional robust/shared scaling and outlier overlays.
 
 If you want to start in a specific grouped view, pass `mode="real"`, `mode="imag"`,
-`mode="phase"`, `mode="sign"`, or `mode="signed_value"` directly in
+`mode="phase"`, `mode="log_magnitude"`, `mode="sparsity"`, `mode="nan_inf"`, `mode="sign"`, or
+`mode="signed_value"` directly in
 `TensorElementsConfig(...)`.
 
 ## Most Common Workflows
@@ -397,7 +405,7 @@ For fuller backend examples, see [docs/backends.md](docs/backends.md).
 | `hover_labels` | Enable hover tooltips in interactive sessions. |
 | `show_contraction_scheme` | Draw contraction-step regions. |
 | `contraction_playback` | Start with playback controls enabled when controls are shown. |
-| `contraction_scheme_cost_hover` | Show cost tooltip on scheme regions. |
+| `contraction_scheme_cost_hover` | Show contraction details in the playback panel. |
 | `tensor_label_refinement` | `"auto"`, `"always"`, or `"never"` for the expensive label-fit pass. |
 | `layout_iterations` | Force-layout effort. |
 | `positions` | Supply custom node coordinates. |

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/README.md

@@ -120,13 +120,14 @@ show_tensor_elements(
 
 When several tensors are present, the figure keeps one tensor active at a time and uses a slider
 to switch between them. The interactive controls are grouped: `basic` (`elements`, `magnitude`,
-`distribution`, `data`), `complex` (`real`, `imag`, `phase`), and `diagnostic` (`sign`,
-`signed_value`).
+`log_magnitude`, `distribution`, `data`), `complex` (`real`, `imag`, `phase`), and
+`diagnostic` (`sign`, `signed_value`, `sparsity`, `nan_inf`).
 
 - `data`: single tensor, iterable of tensors, supported backend-native tensor collections, or an
   `EinsumTrace` with live tensor values.
 - `engine`: optional backend override. If omitted, the library auto-detects it.
 - `config`: tensor-inspection behavior lives here.
+- `data` mode now includes global stats, per-axis summaries, and top-k coordinates by magnitude.
 - `ax`: render a single tensor into an existing Matplotlib axis.
 - `show_controls`: if `True`, add compact Matplotlib `group + mode` controls and, when needed, a tensor slider.
 - `show`: if `False`, nothing is displayed automatically.
@@ -169,6 +170,10 @@ config = TensorElementsConfig(
     mode="auto",
     max_matrix_shape=(256, 256),
     histogram_bins=40,
+    topk_count=8,
+    robust_percentiles=(1.0, 99.0),
+    shared_color_scale=False,
+    highlight_outliers=False,
 )
 ```
 
@@ -177,10 +182,13 @@ This is where you configure:
 - the active inspection mode,
 - row/column axis grouping for rank > 2 tensors,
 - heatmap downsampling limits,
-- histogram sampling and bin count.
+- histogram sampling and bin count,
+- data-summary depth (`topk_count`),
+- optional robust/shared scaling and outlier overlays.
 
 If you want to start in a specific grouped view, pass `mode="real"`, `mode="imag"`,
-`mode="phase"`, `mode="sign"`, or `mode="signed_value"` directly in
+`mode="phase"`, `mode="log_magnitude"`, `mode="sparsity"`, `mode="nan_inf"`, `mode="sign"`, or
+`mode="signed_value"` directly in
 `TensorElementsConfig(...)`.
 
 ## Most Common Workflows
@@ -346,7 +354,7 @@ For fuller backend examples, see [docs/backends.md](docs/backends.md).
 | `hover_labels` | Enable hover tooltips in interactive sessions. |
 | `show_contraction_scheme` | Draw contraction-step regions. |
 | `contraction_playback` | Start with playback controls enabled when controls are shown. |
-| `contraction_scheme_cost_hover` | Show cost tooltip on scheme regions. |
+| `contraction_scheme_cost_hover` | Show contraction details in the playback panel. |
 | `tensor_label_refinement` | `"auto"`, `"always"`, or `"never"` for the expensive label-fit pass. |
 | `layout_iterations` | Force-layout effort. |
 | `positions` | Supply custom node coordinates. |

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensor-network-visualization"
-version = "1.5.1"
+version = "1.5.2"
 description = "Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks."
 authors = [{ name = "Alejandro Mata Ali" }]
 readme = "README.md"
@@ -62,6 +62,10 @@ tensor_network_viz = ["py.typed"]
 testpaths = ["tests"]
 addopts = "-p no:cacheprovider"
 pythonpath = ["src"]
+markers = [
+    "perf: runtime-sensitive regression checks and throughput guards",
+    "smoke: lightweight render smoke checks",
+]
 
 [tool.ruff]
 line-length = 100
@@ -77,6 +81,4 @@ select = ["E", "F", "I", "B", "UP", "C4", "SIM", "RUF100"]
 "examples/**/*.py" = ["E402"]
 # Repo-root bench helpers insert ``src`` on sys.path before third-party imports.
 "scripts/**/*.py" = ["E402"]
-# Draw/layout splits mirror the old monolith: submodule star-import of tuning constants.
-"src/tensor_network_viz/_core/draw/*.py" = ["F403", "F405"]
 "src/tensor_network_viz/_core/layout/body.py" = ["F403", "F405"]

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2/src/tensor_network_visualization.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.5.1
+Version: 1.5.2
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -171,13 +171,14 @@ show_tensor_elements(
 
 When several tensors are present, the figure keeps one tensor active at a time and uses a slider
 to switch between them. The interactive controls are grouped: `basic` (`elements`, `magnitude`,
-`distribution`, `data`), `complex` (`real`, `imag`, `phase`), and `diagnostic` (`sign`,
-`signed_value`).
+`log_magnitude`, `distribution`, `data`), `complex` (`real`, `imag`, `phase`), and
+`diagnostic` (`sign`, `signed_value`, `sparsity`, `nan_inf`).
 
 - `data`: single tensor, iterable of tensors, supported backend-native tensor collections, or an
   `EinsumTrace` with live tensor values.
 - `engine`: optional backend override. If omitted, the library auto-detects it.
 - `config`: tensor-inspection behavior lives here.
+- `data` mode now includes global stats, per-axis summaries, and top-k coordinates by magnitude.
 - `ax`: render a single tensor into an existing Matplotlib axis.
 - `show_controls`: if `True`, add compact Matplotlib `group + mode` controls and, when needed, a tensor slider.
 - `show`: if `False`, nothing is displayed automatically.
@@ -220,6 +221,10 @@ config = TensorElementsConfig(
     mode="auto",
     max_matrix_shape=(256, 256),
     histogram_bins=40,
+    topk_count=8,
+    robust_percentiles=(1.0, 99.0),
+    shared_color_scale=False,
+    highlight_outliers=False,
 )
 ```
 
@@ -228,10 +233,13 @@ This is where you configure:
 - the active inspection mode,
 - row/column axis grouping for rank > 2 tensors,
 - heatmap downsampling limits,
-- histogram sampling and bin count.
+- histogram sampling and bin count,
+- data-summary depth (`topk_count`),
+- optional robust/shared scaling and outlier overlays.
 
 If you want to start in a specific grouped view, pass `mode="real"`, `mode="imag"`,
-`mode="phase"`, `mode="sign"`, or `mode="signed_value"` directly in
+`mode="phase"`, `mode="log_magnitude"`, `mode="sparsity"`, `mode="nan_inf"`, `mode="sign"`, or
+`mode="signed_value"` directly in
 `TensorElementsConfig(...)`.
 
 ## Most Common Workflows
@@ -397,7 +405,7 @@ For fuller backend examples, see [docs/backends.md](docs/backends.md).
 | `hover_labels` | Enable hover tooltips in interactive sessions. |
 | `show_contraction_scheme` | Draw contraction-step regions. |
 | `contraction_playback` | Start with playback controls enabled when controls are shown. |
-| `contraction_scheme_cost_hover` | Show cost tooltip on scheme regions. |
+| `contraction_scheme_cost_hover` | Show contraction details in the playback panel. |
 | `tensor_label_refinement` | `"auto"`, `"always"`, or `"never"` for the expensive label-fit pass. |
 | `layout_iterations` | Force-layout effort. |
 | `positions` | Supply custom node coordinates. |

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/src/tensor_network_visualization.egg-info/SOURCES.txt

@@ -8,7 +8,11 @@ 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/_engine_specs.py
+src/tensor_network_viz/_input_inspection.py
+src/tensor_network_viz/_interactive_scene.py
+src/tensor_network_viz/_matplotlib_state.py
 src/tensor_network_viz/_registry.py
+src/tensor_network_viz/_tensor_elements_controller.py
 src/tensor_network_viz/_tensor_elements_data.py
 src/tensor_network_viz/_tensor_elements_payloads.py
 src/tensor_network_viz/_tensor_elements_support.py
@@ -45,6 +49,7 @@ src/tensor_network_viz/_core/draw/edges.py
 src/tensor_network_viz/_core/draw/fonts_and_scale.py
 src/tensor_network_viz/_core/draw/graph_pipeline.py
 src/tensor_network_viz/_core/draw/hover.py
+src/tensor_network_viz/_core/draw/label_descriptors.py
 src/tensor_network_viz/_core/draw/labels_misc.py
 src/tensor_network_viz/_core/draw/pick_distance.py
 src/tensor_network_viz/_core/draw/plotter.py
@@ -63,6 +68,11 @@ src/tensor_network_viz/_core/layout/geometry.py
 src/tensor_network_viz/_core/layout/parameters.py
 src/tensor_network_viz/_core/layout/positions.py
 src/tensor_network_viz/_core/layout/types.py
+src/tensor_network_viz/_interaction/__init__.py
+src/tensor_network_viz/_interaction/bridge.py
+src/tensor_network_viz/_interaction/controller.py
+src/tensor_network_viz/_interaction/scheme.py
+src/tensor_network_viz/_interaction/state.py
 src/tensor_network_viz/einsum_module/__init__.py
 src/tensor_network_viz/einsum_module/_backend.py
 src/tensor_network_viz/einsum_module/_equation.py
@@ -104,6 +114,7 @@ tests/test_graph_cache.py
 tests/test_integration_einsum.py
 tests/test_integration_tensorkrowch.py
 tests/test_integration_tensornetwork.py
+tests/test_interaction_state.py
 tests/test_label_draw_metrics_perf.py
 tests/test_label_format.py
 tests/test_layout_core.py

tensor_network_visualization-1.5.2/src/tensor_network_viz/_core/_draw_common.py

@@ -0,0 +1,9 @@
+"""Shared scale and style parameters for 2D and 3D drawing."""
+
+from __future__ import annotations
+
+from . import draw as _draw
+
+__all__ = _draw.__all__
+
+globals().update({name: getattr(_draw, name) for name in __all__})

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/src/tensor_network_viz/_core/draw/contraction_edges.py

@@ -13,6 +13,7 @@ from ..layout import NodePositions
 from .constants import _CURVE_NEAR_PAIR_REF, _CURVE_OFFSET_FACTOR
 from .edge_labels import _plot_contraction_index_captions
 from .fonts_and_scale import _DrawScaleParams
+from .label_descriptors import _TextLabelDescriptor
 from .labels_misc import _contraction_hover_label_text
 from .plotter import _PlotAdapter
 from .scene_state import _RenderedEdgeGeometry
@@ -113,6 +114,7 @@ def _draw_contraction_edge_labels(
     ax: Any,
     scale: float,
     zorder_label: float | None = None,
+    label_sink: list[_TextLabelDescriptor] | None = None,
 ) -> None:
     hover_targets = getattr(plotter, "_hover_edge_targets", None)
     if config.hover_labels and hover_targets is not None:
@@ -139,6 +141,7 @@ def _draw_contraction_edge_labels(
         ax=ax,
         scale=scale,
         zorder_label=zorder_label,
+        label_sink=label_sink,
     )
 
 

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/src/tensor_network_viz/_core/draw/dangling_self_edges.py

@@ -10,8 +10,12 @@ from .._label_format import format_tensor_node_label
 from ..curves import _ellipse_points, _ellipse_points_3d, _require_self_endpoints
 from ..graph import _EdgeData, _endpoint_index_caption, _GraphData
 from ..layout import AxisDirections, NodePositions, _orthogonal_unit
-from .constants import *
+from .constants import (
+    _EDGE_INDEX_LABEL_ALONG_FRAC,
+    _PHYS_DANGLING_2D_FRAC_FROM_TIP,
+)
 from .fonts_and_scale import _DrawScaleParams
+from .label_descriptors import _TextLabelDescriptor
 from .labels_misc import (
     _curve_index_outside_disk,
     _dangling_hover_label_text,
@@ -116,6 +120,7 @@ def _draw_dangling_edge_labels(
     ax: Any,
     scale: float,
     zorder_label: float | None = None,
+    label_sink: list[_TextLabelDescriptor] | None = None,
 ) -> None:
     if not edge.label and not config.hover_labels:
         return
@@ -148,6 +153,11 @@ def _draw_dangling_edge_labels(
         return
 
     raw_label = edge.label
+    stub_segment = np.stack([np.asarray(start, dtype=float), np.asarray(end, dtype=float)], axis=0)
+    stub_length = _polyline_arc_length_total(stub_segment)
+    distance_from_tip = float(_PHYS_DANGLING_2D_FRAC_FROM_TIP) * stub_length
+    point, tangent = _point_tangent_along_polyline_from_end(stub_segment, distance_from_tip)
+
     fontsize = _edge_index_fontsize_for_bond(
         raw_label,
         bond_start=start,
@@ -163,10 +173,6 @@ def _draw_dangling_edge_labels(
         bbox_pad=p.index_bbox_pad,
         zorder=zorder_label,
     )
-    stub_segment = np.stack([np.asarray(start, dtype=float), np.asarray(end, dtype=float)], axis=0)
-    stub_length = _polyline_arc_length_total(stub_segment)
-    distance_from_tip = float(_PHYS_DANGLING_2D_FRAC_FROM_TIP) * stub_length
-    point, tangent = _point_tangent_along_polyline_from_end(stub_segment, distance_from_tip)
     if dimensions == 2:
         start_2d = np.asarray(start[:2], dtype=float)
         end_2d = np.asarray(end[:2], dtype=float)
@@ -200,11 +206,18 @@ def _draw_dangling_edge_labels(
             scale=scale,
             fontsize_pt=float(fontsize),
         )
-    plotter.plot_text(
-        label_pos,
-        format_tensor_node_label(raw_label),
-        **{**text_kwargs, **align_kwargs},
-    )
+    formatted = format_tensor_node_label(raw_label)
+    kwargs = {**text_kwargs, **align_kwargs}
+    if label_sink is not None:
+        label_sink.append(
+            _TextLabelDescriptor(
+                position=np.asarray(label_pos, dtype=float).copy(),
+                text=formatted,
+                kwargs=dict(kwargs),
+            )
+        )
+        return
+    plotter.plot_text(label_pos, formatted, **kwargs)
 
 
 def _draw_self_loop_edge(
@@ -299,6 +312,7 @@ def _draw_self_loop_edge_labels(
     ax: Any,
     scale: float,
     zorder_label: float | None = None,
+    label_sink: list[_TextLabelDescriptor] | None = None,
 ) -> None:
     hover_targets = getattr(plotter, "_hover_edge_targets", None)
     if config.hover_labels and hover_targets is not None:
@@ -388,11 +402,22 @@ def _draw_self_loop_edge_labels(
                 dimensions=dimensions,
             ),
         }
-        plotter.plot_text(
-            np.asarray(q_a, dtype=float) + offset_a,
-            format_tensor_node_label(caption_a),
-            **text_kwargs_a,
-        )
+        position_a = np.asarray(q_a, dtype=float) + offset_a
+        formatted_a = format_tensor_node_label(caption_a)
+        if label_sink is not None:
+            label_sink.append(
+                _TextLabelDescriptor(
+                    position=np.asarray(position_a, dtype=float).copy(),
+                    text=formatted_a,
+                    kwargs=dict(text_kwargs_a),
+                )
+            )
+        else:
+            plotter.plot_text(
+                position_a,
+                formatted_a,
+                **text_kwargs_a,
+            )
     if caption_b:
         offset_b = (
             -direction_unit
@@ -430,9 +455,20 @@ def _draw_self_loop_edge_labels(
                 dimensions=dimensions,
             ),
         }
+        formatted_b = format_tensor_node_label(caption_b)
+        position_b = np.asarray(q_b, dtype=float) + offset_b
+        if label_sink is not None:
+            label_sink.append(
+                _TextLabelDescriptor(
+                    position=np.asarray(position_b, dtype=float).copy(),
+                    text=formatted_b,
+                    kwargs=dict(text_kwargs_b),
+                )
+            )
+            return
         plotter.plot_text(
-            np.asarray(q_b, dtype=float) + offset_b,
-            format_tensor_node_label(caption_b),
+            position_b,
+            formatted_b,
             **text_kwargs_b,
         )
 

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/src/tensor_network_viz/_core/draw/edge_labels.py

@@ -16,6 +16,7 @@ from ..graph import (
 )
 from ..layout import NodePositions
 from .fonts_and_scale import _DrawScaleParams
+from .label_descriptors import _TextLabelDescriptor
 from .labels_misc import _edge_index_text_kwargs
 from .plotter import _PlotAdapter
 from .viewport_geometry import (
@@ -46,6 +47,7 @@ def _plot_contraction_index_captions(
     ax: Any,
     scale: float,
     zorder_label: float | None = None,
+    label_sink: list[_TextLabelDescriptor] | None = None,
 ) -> None:
     ep_l, ep_r = _require_contraction_endpoints(edge)
     cap_l: str | None = _endpoint_index_caption(ep_l, edge, graph)
@@ -129,11 +131,18 @@ def _plot_contraction_index_captions(
                 scale=scale,
                 fontsize_pt=float(fontsize),
             )
-        plotter.plot_text(
-            position,
-            format_tensor_node_label(cap),
-            **{**text_kwargs, **align_kwargs},
-        )
+        formatted = format_tensor_node_label(cap)
+        kwargs = {**text_kwargs, **align_kwargs}
+        if label_sink is not None:
+            label_sink.append(
+                _TextLabelDescriptor(
+                    position=np.asarray(position, dtype=float).copy(),
+                    text=formatted,
+                    kwargs=dict(kwargs),
+                )
+            )
+            continue
+        plotter.plot_text(position, formatted, **kwargs)
 
 
 __all__ = ["_plot_contraction_index_captions"]

{tensor_network_visualization-1.5.1 → tensor_network_visualization-1.5.2}/src/tensor_network_viz/_core/draw/edge_orchestration.py

@@ -7,7 +7,13 @@ from ...config import PlotConfig
 from ..contractions import _ContractionGroups
 from ..graph import _EdgeData, _GraphData
 from ..layout import AxisDirections, NodePositions
-from .constants import *
+from .constants import (
+    _ZORDER_LAYER_BASE,
+    _ZORDER_LAYER_BOND,
+    _ZORDER_LAYER_DANGLING,
+    _ZORDER_LAYER_EDGE_INDEX,
+    _ZORDER_LAYER_STRIDE,
+)
 from .contraction_edges import _draw_contraction_edge
 from .dangling_self_edges import _draw_dangling_edge, _draw_self_loop_edge
 from .fonts_and_scale import _DrawScaleParams