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

{tensor_network_visualization-1.6.3 → 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-1.6.3/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: 1.6.3
+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
@@ -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
@@ -206,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-1.6.3 → tensor_network_visualization-2.0.1}/README.md

@@ -1,18 +1,67 @@
-# 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.
-
-## 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:
 
@@ -152,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-1.6.3 → tensor_network_visualization-2.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensor-network-visualization"
-version = "1.6.3"
+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"
@@ -15,15 +15,19 @@ keywords = ["tensor-network", "visualization", "pytorch", "numpy", "tensorkrowch
 dependencies = [
     "matplotlib>=3.7",
     "networkx>=3.0",
-    "numpy",
+    "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]

{tensor_network_visualization-1.6.3 → 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: 1.6.3
+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
@@ -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
@@ -206,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-1.6.3 → tensor_network_visualization-2.0.1}/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.1}/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.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-1.6.3 → 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-1.6.3 → 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",
+]