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

{tensor_network_visualization-2.0.0/src/tensor_network_visualization.egg-info → tensor_network_visualization-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 2.0.0
+Version: 2.0.2
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -73,8 +73,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 +123,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`
@@ -259,12 +263,23 @@ 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
   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.0 → tensor_network_visualization-2.0.2}/README.md

@@ -18,8 +18,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
@@ -63,15 +62,20 @@ Tensor-network libraries expose different Python objects. This package gives the
 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:
-
-```python
-show_tensor_network(...)
-show_tensor_elements(...)
-show_tensor_comparison(...)
-normalize_tensor_network(...)
-export_tensor_network_snapshot(...)
-```
+The common entry points are:
+
+```python
+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
 
@@ -201,16 +205,27 @@ 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:
+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
-  disconnected networks.
-
-For batch checks, use:
+- `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
 python examples/run_all_examples.py --group engines --views 2d --list

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.2}/pyproject.toml

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensor-network-visualization
-Version: 2.0.0
+Version: 2.0.2
 Summary: Minimal Matplotlib visualizations for TensorKrowch, TensorNetwork, Quimb, TeNPy, and traced PyTorch/NumPy einsum tensor networks.
 Author: Alejandro Mata Ali
 License-Expression: MIT
@@ -73,8 +73,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 +123,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`
@@ -259,12 +263,23 @@ 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
   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.0 → tensor_network_visualization-2.0.2}/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
@@ -162,4 +165,5 @@ 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.0 → tensor_network_visualization-2.0.2}/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
@@ -48,7 +50,12 @@ 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 .translation import translate_tensor_network
     from .viewer import show_tensor_network
 else:
 
@@ -148,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"),
@@ -163,12 +187,14 @@ _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"),
     "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"),
@@ -217,11 +243,13 @@ __all__ = [
     "NormalizedTensorNode",
     "PlotConfig",
     "PlotTheme",
+    "TranslationTargetName",
     "TensorComparisonConfig",
     "TensorAnalysisConfig",
     "TensorDataError",
     "TensorDataTypeError",
     "TensorElementsConfig",
+    "TensorElementsTheme",
     "TensorNetworkDiagnosticsConfig",
     "TensorNetworkFocus",
     "TensorNetworkLayoutSnapshot",
@@ -242,6 +270,7 @@ __all__ = [
     "show_tensor_comparison",
     "show_tensor_elements",
     "show_tensor_network",
+    "translate_tensor_network",
 ]
 
 _ = _package_logging

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.2}/src/tensor_network_viz/_engine_specs.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from typing import Literal, TypeAlias
 
 EngineName: TypeAlias = Literal["tensorkrowch", "tensornetwork", "quimb", "tenpy", "einsum"]
+TranslationTargetName: TypeAlias = Literal["tensorkrowch", "tensornetwork", "quimb", "einsum"]
 
 ENGINE_MODULE_MAP: dict[EngineName, tuple[str, str, str]] = {
     "tensorkrowch": (

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.2}/src/tensor_network_viz/_interaction/controller.py

@@ -298,6 +298,7 @@ class _InteractiveTensorFigureController:
                 ),
                 node_records_by_name=self._node_records_by_name,
                 placeholder_engine=engine,
+                inspector_config=config.tensor_inspector_config,
                 on_closed=self._on_tensor_inspector_closed,
             )
 

{tensor_network_visualization-2.0.0 → tensor_network_visualization-2.0.2}/src/tensor_network_viz/_interaction/tensor_inspector.py

@@ -191,6 +191,7 @@ class _LinkedTensorInspectorController:
         step_records: tuple[_PlaybackStepRecord, ...] | None,
         node_records_by_name: dict[str, _TensorRecord] | None,
         placeholder_engine: EngineName,
+        inspector_config: TensorElementsConfig | None,
         on_closed: Callable[[], None],
     ) -> None:
         self._step_records = tuple(step_records or ())
@@ -199,7 +200,40 @@ class _LinkedTensorInspectorController:
         )
         self._placeholder_engine: EngineName = placeholder_engine
         self._on_closed = on_closed
-        self._config = TensorElementsConfig(figsize=_INSPECTOR_FIGSIZE)
+        base_config = TensorElementsConfig() if inspector_config is None else inspector_config
+        self._config = TensorElementsConfig(
+            mode=base_config.mode,
+            row_axes=base_config.row_axes,
+            col_axes=base_config.col_axes,
+            analysis=base_config.analysis,
+            theme=base_config.theme,
+            continuous_cmap=base_config.continuous_cmap,
+            log_magnitude_cmap=base_config.log_magnitude_cmap,
+            phase_cmap=base_config.phase_cmap,
+            diverging_cmap=base_config.diverging_cmap,
+            sign_colors=base_config.sign_colors,
+            sparsity_colors=base_config.sparsity_colors,
+            nan_inf_colors=base_config.nan_inf_colors,
+            series_color=base_config.series_color,
+            histogram_color=base_config.histogram_color,
+            histogram_edge_color=base_config.histogram_edge_color,
+            zero_marker_color=base_config.zero_marker_color,
+            hover_facecolor=base_config.hover_facecolor,
+            hover_edgecolor=base_config.hover_edgecolor,
+            summary_facecolor=base_config.summary_facecolor,
+            summary_edgecolor=base_config.summary_edgecolor,
+            figsize=_INSPECTOR_FIGSIZE,
+            max_matrix_shape=base_config.max_matrix_shape,
+            shared_color_scale=base_config.shared_color_scale,
+            robust_percentiles=base_config.robust_percentiles,
+            highlight_outliers=base_config.highlight_outliers,
+            outlier_zscore=base_config.outlier_zscore,
+            zero_threshold=base_config.zero_threshold,
+            log_magnitude_floor=base_config.log_magnitude_floor,
+            histogram_bins=base_config.histogram_bins,
+            histogram_max_samples=base_config.histogram_max_samples,
+            topk_count=base_config.topk_count,
+        )
         self._enabled: bool = False
         self._viewer: _StepPlaybackViewer | None = None
         self._figure: Figure | None = None