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

{tensor_network_visualization-1.6.2 → tensor_network_visualization-2.0.0}/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-2.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 1.6.2
+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.2 → 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.2 → 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.2"
+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.2 → 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.2
+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.2 → tensor_network_visualization-2.0.0}/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
@@ -125,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.2 → 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-2.0.0/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-2.0.0}/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-2.0.0}/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",
+]