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

{tensor_network_visualization-2.0.1 → tensor_network_visualization-2.0.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-2.0.1/src/tensor_network_visualization.egg-info → tensor_network_visualization-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 2.0.1
+Version: 2.0.3
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -27,10 +27,14 @@ Requires-Dist: matplotlib>=3.7
 Requires-Dist: networkx>=3.0
 Requires-Dist: numpy>=2.0
 Provides-Extra: dev
+Requires-Dist: bandit; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: ruff; extra == "dev"
+Requires-Dist: setuptools; extra == "dev"
 Requires-Dist: pyright; extra == "dev"
+Requires-Dist: pip-audit; extra == "dev"
+Requires-Dist: twine; extra == "dev"
 Requires-Dist: ipython; extra == "dev"
 Requires-Dist: tensorkrowch; extra == "dev"
 Requires-Dist: tensornetwork; extra == "dev"
@@ -73,8 +77,7 @@ 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.
+These are exports generated with the library and the repository demos.
 
 <p align="center">
   <img
@@ -124,10 +127,15 @@ The common entry points are:
 show_tensor_network(...)
 show_tensor_elements(...)
 show_tensor_comparison(...)
+translate_tensor_network(...)
 normalize_tensor_network(...)
 export_tensor_network_snapshot(...)
 ```
 
+If you want code for another backend rather than a figure, `translate_tensor_network(...)` can turn
+supported TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced `einsum` inputs into executable
+Python for `tensornetwork`, `quimb`, `einsum`, or `tensorkrowch`.
+
 ## Install
 
 - PyPI package name: `tensor-network-visualization`
@@ -267,6 +275,15 @@ plus three focused groups:
 - `geometry`: renders larger irregular, incomplete, triangular, pyramidal, circular, and
   disconnected networks.
 
+There is also a dedicated translation demo:
+
+```bash
+python examples/translate_demo.py --source-engine tensornetwork --target-engine quimb --example mps
+```
+
+It generates Python code for the translated tensor network and can render the original and
+translated versions side by side for comparison.
+
 For batch checks, use:
 
 ```bash

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

@@ -1,67 +1,66 @@
-<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.
+
+<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:
 
@@ -69,10 +68,15 @@ The common entry points are:
 show_tensor_network(...)
 show_tensor_elements(...)
 show_tensor_comparison(...)
+translate_tensor_network(...)
 normalize_tensor_network(...)
 export_tensor_network_snapshot(...)
 ```
 
+If you want code for another backend rather than a figure, `translate_tensor_network(...)` can turn
+supported TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced `einsum` inputs into executable
+Python for `tensornetwork`, `quimb`, `einsum`, or `tensorkrowch`.
+
 ## Install
 
 - PyPI package name: `tensor-network-visualization`
@@ -201,16 +205,25 @@ 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 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.
+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.
+
+There is also a dedicated translation demo:
+
+```bash
+python examples/translate_demo.py --source-engine tensornetwork --target-engine quimb --example mps
+```
+
+It generates Python code for the translated tensor network and can render the original and
+translated versions side by side for comparison.
 
 For batch checks, use:
 

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tensor-network-visualization"
-version = "2.0.1"
+version = "2.0.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"
@@ -39,10 +39,14 @@ Issues = "https://github.com/DOKOS-TAYOS/Tensor-Network-Visualization/issues"
 
 [project.optional-dependencies]
 dev = [
+    "bandit",
     "build",
     "pytest",
     "ruff",
+    "setuptools",
     "pyright",
+    "pip-audit",
+    "twine",
     "ipython",
     "tensorkrowch",
     "tensornetwork",

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 2.0.1
+Version: 2.0.3
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -27,10 +27,14 @@ Requires-Dist: matplotlib>=3.7
 Requires-Dist: networkx>=3.0
 Requires-Dist: numpy>=2.0
 Provides-Extra: dev
+Requires-Dist: bandit; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: ruff; extra == "dev"
+Requires-Dist: setuptools; extra == "dev"
 Requires-Dist: pyright; extra == "dev"
+Requires-Dist: pip-audit; extra == "dev"
+Requires-Dist: twine; extra == "dev"
 Requires-Dist: ipython; extra == "dev"
 Requires-Dist: tensorkrowch; extra == "dev"
 Requires-Dist: tensornetwork; extra == "dev"
@@ -73,8 +77,7 @@ 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.
+These are exports generated with the library and the repository demos.
 
 <p align="center">
   <img
@@ -124,10 +127,15 @@ The common entry points are:
 show_tensor_network(...)
 show_tensor_elements(...)
 show_tensor_comparison(...)
+translate_tensor_network(...)
 normalize_tensor_network(...)
 export_tensor_network_snapshot(...)
 ```
 
+If you want code for another backend rather than a figure, `translate_tensor_network(...)` can turn
+supported TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced `einsum` inputs into executable
+Python for `tensornetwork`, `quimb`, `einsum`, or `tensorkrowch`.
+
 ## Install
 
 - PyPI package name: `tensor-network-visualization`
@@ -267,6 +275,15 @@ plus three focused groups:
 - `geometry`: renders larger irregular, incomplete, triangular, pyramidal, circular, and
   disconnected networks.
 
+There is also a dedicated translation demo:
+
+```bash
+python examples/translate_demo.py --source-engine tensornetwork --target-engine quimb --example mps
+```
+
+It generates Python code for the translated tensor network and can render the original and
+translated versions side by side for comparison.
+
 For batch checks, use:
 
 ```bash

{tensor_network_visualization-2.0.1 → tensor_network_visualization-2.0.3}/src/tensor_network_visualization.egg-info/SOURCES.txt

@@ -25,6 +25,8 @@ src/tensor_network_viz/_tensor_elements_models.py
 src/tensor_network_viz/_tensor_elements_payloads.py
 src/tensor_network_viz/_tensor_elements_rendering.py
 src/tensor_network_viz/_tensor_elements_support.py
+src/tensor_network_viz/_translation_codegen.py
+src/tensor_network_viz/_translation_models.py
 src/tensor_network_viz/_typing.py
 src/tensor_network_viz/_ui_utils.py
 src/tensor_network_viz/_widgets.py
@@ -38,6 +40,7 @@ src/tensor_network_viz/tensor_comparison.py
 src/tensor_network_viz/tensor_comparison_config.py
 src/tensor_network_viz/tensor_elements.py
 src/tensor_network_viz/tensor_elements_config.py
+src/tensor_network_viz/translation.py
 src/tensor_network_viz/viewer.py
 src/tensor_network_viz/_core/__init__.py
 src/tensor_network_viz/_core/_draw_common.py
@@ -156,10 +159,12 @@ tests/test_plotting.py
 tests/test_public_api.py
 tests/test_quimb_backend.py
 tests/test_render_performance_controls.py
+tests/test_security_config.py
 tests/test_show_tensor_network_throughput.py
 tests/test_snapshot_api.py
 tests/test_tenpy_backend.py
 tests/test_tensor_comparison.py
 tests/test_tensor_elements.py
 tests/test_tensor_elements_perf.py
+tests/test_translation_api.py
 tests/test_verify_script.py

{tensor_network_visualization-2.0.1 → tensor_network_visualization-2.0.3}/src/tensor_network_visualization.egg-info/requires.txt

@@ -3,10 +3,14 @@ networkx>=3.0
 numpy>=2.0
 
 [dev]
+bandit
 build
 pytest
 ruff
+setuptools
 pyright
+pip-audit
+twine
 ipython
 tensorkrowch
 tensornetwork

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

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from importlib import import_module
+from os import PathLike
 from typing import TYPE_CHECKING, Any
 
 from . import _logging as _package_logging
@@ -19,6 +20,7 @@ if TYPE_CHECKING:
         PlotTheme,
         TensorNetworkDiagnosticsConfig,
         TensorNetworkFocus,
+        TranslationTargetName,
         ViewName,
     )
     from .contraction_viewer import ContractionViewer2D, ContractionViewer3D
@@ -53,6 +55,7 @@ if TYPE_CHECKING:
         TensorElementsConfig,
         TensorElementsTheme,
     )
+    from .translation import translate_tensor_network
     from .viewer import show_tensor_network
 else:
 
@@ -152,6 +155,23 @@ else:
             show=show,
         )
 
+    def translate_tensor_network(
+        network: Any,
+        *,
+        engine: EngineName | None = None,
+        target_engine: TranslationTargetName,
+        path: str | PathLike[str] | None = None,
+    ) -> str:
+        """Lazily dispatch to :func:`tensor_network_viz.translation.translate_tensor_network`."""
+        from .translation import translate_tensor_network as _translate_tensor_network
+
+        return _translate_tensor_network(
+            network,
+            engine=engine,
+            target_engine=target_engine,
+            path=path,
+        )
+
 
 _LAZY_EXPORTS: dict[str, tuple[str, str]] = {
     "AxisConfigurationError": (".exceptions", "AxisConfigurationError"),
@@ -167,6 +187,7 @@ _LAZY_EXPORTS: dict[str, tuple[str, str]] = {
     "NormalizedTensorNode": (".snapshot", "NormalizedTensorNode"),
     "PlotConfig": (".config", "PlotConfig"),
     "PlotTheme": (".config", "PlotTheme"),
+    "TranslationTargetName": ("._engine_specs", "TranslationTargetName"),
     "TenPyTensorNetwork": (".tenpy.explicit", "TenPyTensorNetwork"),
     "TensorComparisonConfig": (".tensor_comparison_config", "TensorComparisonConfig"),
     "TensorAnalysisConfig": (".tensor_elements_config", "TensorAnalysisConfig"),
@@ -222,6 +243,7 @@ __all__ = [
     "NormalizedTensorNode",
     "PlotConfig",
     "PlotTheme",
+    "TranslationTargetName",
     "TensorComparisonConfig",
     "TensorAnalysisConfig",
     "TensorDataError",
@@ -248,6 +270,7 @@ __all__ = [
     "show_tensor_comparison",
     "show_tensor_elements",
     "show_tensor_network",
+    "translate_tensor_network",
 ]
 
 _ = _package_logging

{tensor_network_visualization-2.0.1 → tensor_network_visualization-2.0.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-2.0.1 → tensor_network_visualization-2.0.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",
+]