PyPI - ttnn-visualizer - Versions diffs - 0.49.0__py3-none-any.whl → 0.64.0__py3-none-any.whl - Mend

ttnn-visualizer 0.49.0py3-none-any.whl → 0.64.0py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (36) hide show

ttnn_visualizer/app.py +151 -49
ttnn_visualizer/csv_queries.py +154 -45
ttnn_visualizer/decorators.py +0 -9
ttnn_visualizer/exceptions.py +0 -7
ttnn_visualizer/models.py +20 -1
ttnn_visualizer/queries.py +8 -0
ttnn_visualizer/serializers.py +53 -9
ttnn_visualizer/settings.py +24 -10
ttnn_visualizer/ssh_client.py +1 -4
ttnn_visualizer/static/assets/allPaths-DWjqav_8.js +1 -0
ttnn_visualizer/static/assets/allPathsLoader-B0eRT9aL.js +2 -0
ttnn_visualizer/static/assets/index-BE2R-cuu.css +1 -0
ttnn_visualizer/static/assets/index-BZITDwoa.js +1 -0
ttnn_visualizer/static/assets/{index-DVrPLQJ7.js → index-DDrUX09k.js} +274 -479
ttnn_visualizer/static/assets/index-voJy5fZe.js +1 -0
ttnn_visualizer/static/assets/splitPathsBySizeLoader-_GpmIkFm.js +1 -0
ttnn_visualizer/static/index.html +2 -2
ttnn_visualizer/tests/test_serializers.py +2 -0
ttnn_visualizer/tests/test_utils.py +362 -0
ttnn_visualizer/utils.py +142 -0
ttnn_visualizer/views.py +181 -87
{ttnn_visualizer-0.49.0.dist-info → ttnn_visualizer-0.64.0.dist-info}/METADATA +58 -30
ttnn_visualizer-0.64.0.dist-info/RECORD +44 -0
{ttnn_visualizer-0.49.0.dist-info → ttnn_visualizer-0.64.0.dist-info}/licenses/LICENSE +6 -0
ttnn_visualizer/remote_sqlite_setup.py +0 -100
ttnn_visualizer/static/assets/allPaths-G_CNx_x1.js +0 -1
ttnn_visualizer/static/assets/allPathsLoader-s_Yfmxfp.js +0 -2
ttnn_visualizer/static/assets/index-CnPrfHYh.js +0 -1
ttnn_visualizer/static/assets/index-Cnc1EkDo.js +0 -1
ttnn_visualizer/static/assets/index-UuXdrHif.css +0 -7
ttnn_visualizer/static/assets/splitPathsBySizeLoader-ivxxaHxa.js +0 -1
ttnn_visualizer-0.49.0.dist-info/RECORD +0 -44
{ttnn_visualizer-0.49.0.dist-info → ttnn_visualizer-0.64.0.dist-info}/WHEEL +0 -0
{ttnn_visualizer-0.49.0.dist-info → ttnn_visualizer-0.64.0.dist-info}/entry_points.txt +0 -0
{ttnn_visualizer-0.49.0.dist-info → ttnn_visualizer-0.64.0.dist-info}/licenses/LICENSE_understanding.txt +0 -0
{ttnn_visualizer-0.49.0.dist-info → ttnn_visualizer-0.64.0.dist-info}/top_level.txt +0 -0

ttnn_visualizer/static/assets/splitPathsBySizeLoader-_GpmIkFm.js ADDED Viewed

	@@ -0,0 +1 @@
1	+ import{p as r,I as s,_ as a}from"./index-DDrUX09k.js";const n=async(o,_)=>{const i=r(o);let t;return _===s.STANDARD?t=await a(()=>import("./index-voJy5fZe.js").then(e=>e.I),[]):t=await a(()=>import("./index-BZITDwoa.js").then(e=>e.I),[]),t[i]};export{n as splitPathsBySizeLoader};

ttnn_visualizer/static/index.html CHANGED Viewed

@@ -34,8 +34,8 @@
             /* SERVER_CONFIG */
         </script>
-      <script type="module" crossorigin src="/static/assets/index-DVrPLQJ7.js"></script>
-      <link rel="stylesheet" crossorigin href="/static/assets/index-UuXdrHif.css">
+      <script type="module" crossorigin src="/static/assets/index-DDrUX09k.js"></script>
+      <link rel="stylesheet" crossorigin href="/static/assets/index-BE2R-cuu.css">
     </head>
     <body>

ttnn_visualizer/tests/test_serializers.py CHANGED Viewed

@@ -134,6 +134,7 @@ class TestSerializers(unittest.TestCase):
                         "device_addresses": [25],
                     }
                 ],
+                "error": None,
             }
         ]
@@ -447,6 +448,7 @@ class TestSerializers(unittest.TestCase):
                 }
             ],
             "stack_trace": "trace1",
+            "error": None,
         }
         self.assertEqual(result, expected)

ttnn_visualizer/tests/test_utils.py ADDED Viewed

@@ -0,0 +1,362 @@
+# SPDX-License-Identifier: Apache-2.0
+#
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
+from unittest.mock import mock_open, patch
+from ttnn_visualizer.utils import (
+    find_gunicorn_path,
+    get_app_data_directory,
+    is_running_in_container,
+)
+@patch("sys.argv", ["/home/user/.local/bin/ttnn-visualizer"])
+@patch("os.access")
+@patch("pathlib.Path.exists")
+@patch("pathlib.Path.is_file")
+@patch("shutil.which")
+def test_find_gunicorn_in_same_directory(
+    mock_which, mock_is_file, mock_exists, mock_access
+):
+    """Test finding gunicorn in the same directory as ttnn-visualizer."""
+    mock_exists.return_value = True
+    mock_is_file.return_value = True
+    mock_access.return_value = True
+    mock_which.return_value = None  # Not in PATH
+    gunicorn_path, warning = find_gunicorn_path()
+    assert gunicorn_path.endswith("/home/user/.local/bin/gunicorn")
+    assert warning is None
+@patch("sys.argv", ["/home/user/.local/bin/ttnn-visualizer"])
+@patch("os.access")
+@patch("pathlib.Path.exists")
+@patch("pathlib.Path.is_file")
+@patch("shutil.which")
+def test_find_multiple_gunicorn_installations(
+    mock_which, mock_is_file, mock_exists, mock_access
+):
+    """Test warning when multiple gunicorn installations are detected."""
+    mock_exists.return_value = True
+    mock_is_file.return_value = True
+    mock_access.return_value = True
+    mock_which.return_value = "/usr/bin/gunicorn"  # Different one in PATH
+    gunicorn_path, warning = find_gunicorn_path()
+    assert gunicorn_path.endswith("/home/user/.local/bin/gunicorn")
+    assert warning is not None
+    assert "Multiple gunicorn installations detected" in warning
+@patch("sys.argv", ["/home/user/.local/bin/ttnn-visualizer"])
+@patch("os.access")
+@patch("pathlib.Path.exists")
+@patch("pathlib.Path.is_file")
+@patch("shutil.which")
+def test_gunicorn_not_executable(mock_which, mock_is_file, mock_exists, mock_access):
+    """Test when gunicorn exists but is not executable."""
+    mock_exists.return_value = True
+    mock_is_file.return_value = True
+    mock_access.return_value = False  # Not executable
+    mock_which.return_value = "/usr/bin/gunicorn"
+    gunicorn_path, warning = find_gunicorn_path()
+    assert gunicorn_path == "/usr/bin/gunicorn"
+    assert warning is not None
+    assert "not executable" in warning
+    assert "chmod +x" in warning
+@patch("sys.argv", ["/home/user/.local/bin/ttnn-visualizer"])
+@patch("os.access")
+@patch("pathlib.Path.exists")
+@patch("pathlib.Path.is_file")
+@patch("shutil.which")
+def test_fallback_to_path(mock_which, mock_is_file, mock_exists, mock_access):
+    """Test falling back to PATH when not in same directory."""
+    mock_exists.return_value = False
+    mock_is_file.return_value = False
+    mock_which.return_value = "/usr/bin/gunicorn"
+    gunicorn_path, warning = find_gunicorn_path()
+    assert gunicorn_path == "/usr/bin/gunicorn"
+    assert warning is not None
+    assert "not found in" in warning
+    assert "Falling back" in warning
+@patch("sys.argv", ["/home/user/.local/bin/ttnn-visualizer"])
+@patch("os.access")
+@patch("pathlib.Path.exists")
+@patch("pathlib.Path.is_file")
+@patch("shutil.which")
+def test_gunicorn_not_found(mock_which, mock_is_file, mock_exists, mock_access):
+    """Test when gunicorn is not found anywhere."""
+    mock_exists.return_value = False
+    mock_is_file.return_value = False
+    mock_which.return_value = None
+    gunicorn_path, warning = find_gunicorn_path()
+    assert gunicorn_path == "gunicorn"
+    assert warning is not None
+    assert "ERROR" in warning
+    assert "not found" in warning
+# Tests for is_running_in_container()
+@patch("os.path.exists")
+@patch("os.getenv")
+def test_container_detection_via_dockerenv(mock_getenv, mock_exists):
+    """Test container detection via /.dockerenv file."""
+    mock_exists.return_value = True
+    mock_getenv.return_value = None
+    result = is_running_in_container()
+    assert result is True
+    mock_exists.assert_called_once_with("/.dockerenv")
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/docker/abc123\n11:cpuset:/docker/abc123",
+)
+@patch("os.getenv")
+def test_container_detection_via_cgroup_docker(mock_getenv, mock_file, mock_exists):
+    """Test container detection via /proc/self/cgroup containing 'docker'."""
+    mock_exists.return_value = False  # No /.dockerenv
+    mock_getenv.return_value = None
+    result = is_running_in_container()
+    assert result is True
+    mock_file.assert_called_once_with("/proc/self/cgroup", "r")
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/containerd/abc123\n11:cpuset:/containerd/abc123",
+)
+@patch("os.getenv")
+def test_container_detection_via_cgroup_containerd(mock_getenv, mock_file, mock_exists):
+    """Test container detection via /proc/self/cgroup containing 'containerd'."""
+    mock_exists.return_value = False
+    mock_getenv.return_value = None
+    result = is_running_in_container()
+    assert result is True
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/lxc/container123\n11:cpuset:/lxc/container123",
+)
+@patch("os.getenv")
+def test_container_detection_via_cgroup_lxc(mock_getenv, mock_file, mock_exists):
+    """Test container detection via /proc/self/cgroup containing 'lxc'."""
+    mock_exists.return_value = False
+    mock_getenv.return_value = None
+    result = is_running_in_container()
+    assert result is True
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/kubepods/besteffort/pod123\n11:cpuset:/kubepods/besteffort/pod123",
+)
+@patch("os.getenv")
+def test_container_detection_via_cgroup_kubepods(mock_getenv, mock_file, mock_exists):
+    """Test container detection via /proc/self/cgroup containing 'kubepods'."""
+    mock_exists.return_value = False
+    mock_getenv.return_value = None
+    result = is_running_in_container()
+    assert result is True
+@patch("os.path.exists")
+@patch("builtins.open", side_effect=FileNotFoundError())
+@patch("os.getenv")
+def test_container_detection_cgroup_file_not_found(mock_getenv, mock_file, mock_exists):
+    """Test container detection handles FileNotFoundError from /proc/self/cgroup."""
+    mock_exists.return_value = False
+    def getenv_side_effect(key):
+        return None
+    mock_getenv.side_effect = getenv_side_effect
+    result = is_running_in_container()
+    assert result is False
+@patch("os.path.exists")
+@patch("builtins.open", side_effect=PermissionError())
+@patch("os.getenv")
+def test_container_detection_cgroup_permission_error(
+    mock_getenv, mock_file, mock_exists
+):
+    """Test container detection handles PermissionError from /proc/self/cgroup."""
+    mock_exists.return_value = False
+    def getenv_side_effect(key):
+        return None
+    mock_getenv.side_effect = getenv_side_effect
+    result = is_running_in_container()
+    assert result is False
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/user.slice\n11:cpuset:/",
+)
+def test_container_detection_via_kubernetes_service_host(mock_file, mock_exists):
+    """Test container detection via KUBERNETES_SERVICE_HOST environment variable."""
+    mock_exists.return_value = False
+    with patch.dict("os.environ", {"KUBERNETES_SERVICE_HOST": "10.0.0.1"}, clear=True):
+        result = is_running_in_container()
+    assert result is True
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/user.slice\n11:cpuset:/",
+)
+def test_container_detection_via_kubernetes_port(mock_file, mock_exists):
+    """Test container detection via KUBERNETES_PORT environment variable."""
+    mock_exists.return_value = False
+    with patch.dict(
+        "os.environ", {"KUBERNETES_PORT": "tcp://10.0.0.1:443"}, clear=True
+    ):
+        result = is_running_in_container()
+    assert result is True
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/user.slice\n11:cpuset:/",
+)
+def test_container_detection_via_container_env(mock_file, mock_exists):
+    """Test container detection via 'container' environment variable."""
+    mock_exists.return_value = False
+    with patch.dict("os.environ", {"container": "podman"}, clear=True):
+        result = is_running_in_container()
+    assert result is True
+@patch("os.path.exists")
+@patch(
+    "builtins.open",
+    new_callable=mock_open,
+    read_data="12:pids:/user.slice\n11:cpuset:/",
+)
+@patch("os.getenv")
+def test_no_container_detection(mock_getenv, mock_file, mock_exists):
+    """Test that no container is detected when all checks fail."""
+    mock_exists.return_value = False  # No /.dockerenv
+    mock_getenv.return_value = None  # No container env vars
+    result = is_running_in_container()
+    assert result is False
+# Tests for get_app_data_directory()
+def test_get_app_data_directory_with_tt_metal_home():
+    """Test that get_app_data_directory returns correct path when tt_metal_home is provided."""
+    tt_metal_home = "/path/to/tt-metal"
+    application_dir = "/default/app/dir"
+    result = get_app_data_directory(tt_metal_home, application_dir)
+    assert result == "/path/to/tt-metal/generated/ttnn-visualizer"
+def test_get_app_data_directory_with_none():
+    """Test that get_app_data_directory returns application_dir when tt_metal_home is None."""
+    tt_metal_home = None
+    application_dir = "/default/app/dir"
+    result = get_app_data_directory(tt_metal_home, application_dir)
+    assert result == "/default/app/dir"
+def test_get_app_data_directory_with_empty_string():
+    """Test that get_app_data_directory treats empty string as falsy and returns application_dir."""
+    tt_metal_home = ""
+    application_dir = "/default/app/dir"
+    result = get_app_data_directory(tt_metal_home, application_dir)
+    assert result == "/default/app/dir"
+def test_get_app_data_directory_with_special_characters():
+    """Test that get_app_data_directory handles paths with special characters correctly."""
+    tt_metal_home = "/path/with spaces/and-dashes/tt-metal"
+    application_dir = "/default/app/dir"
+    result = get_app_data_directory(tt_metal_home, application_dir)
+    assert result == "/path/with spaces/and-dashes/tt-metal/generated/ttnn-visualizer"
+def test_get_app_data_directory_with_relative_path():
+    """Test that get_app_data_directory handles relative paths correctly."""
+    tt_metal_home = "../relative/path/tt-metal"
+    application_dir = "/default/app/dir"
+    result = get_app_data_directory(tt_metal_home, application_dir)
+    assert result == "../relative/path/tt-metal/generated/ttnn-visualizer"
+def test_get_app_data_directory_with_trailing_slash():
+    """Test that get_app_data_directory handles paths with trailing slashes correctly."""
+    tt_metal_home = "/path/to/tt-metal/"
+    application_dir = "/default/app/dir"
+    result = get_app_data_directory(tt_metal_home, application_dir)
+    # Path.join handles trailing slashes correctly
+    assert result == "/path/to/tt-metal/generated/ttnn-visualizer"

ttnn_visualizer/utils.py CHANGED Viewed

@@ -6,7 +6,10 @@ import dataclasses
 import enum
 import json
 import logging
+import os
 import re
+import shutil
+import sys
 import time
 from functools import wraps
 from pathlib import Path
@@ -18,6 +21,98 @@ logger = logging.getLogger(__name__)
 LAST_SYNCED_FILE_NAME = ".last-synced"
+def get_app_data_directory(tt_metal_home: Optional[str], application_dir: str) -> str:
+    """
+    Calculate the APP_DATA_DIRECTORY based on TT_METAL_HOME or fallback to application_dir.
+    Args:
+        tt_metal_home: Path to TT-Metal home directory, or None
+        application_dir: Fallback application directory path
+    Returns:
+        Path to the app data directory
+    """
+    if tt_metal_home and tt_metal_home.strip():
+        return str(Path(tt_metal_home).expanduser() / "generated" / "ttnn-visualizer")
+    return application_dir
+def find_gunicorn_path() -> tuple[str, Optional[str]]:
+    """
+    Find the gunicorn executable, prioritizing the same bin directory as ttnn-visualizer.
+    Returns:
+        tuple: (gunicorn_path, warning_message)
+            - gunicorn_path: Full path to the gunicorn executable to use
+            - warning_message: Warning message if there are any issues finding gunicorn
+              (e.g., multiple installations, falling back to PATH, or not found),
+              or None if found without conflicts.
+    """
+    # Get the directory where ttnn-visualizer was run from
+    ttnn_visualizer_path = Path(sys.argv[0]).resolve()
+    bin_dir = ttnn_visualizer_path.parent
+    # Look for gunicorn in the same directory
+    expected_gunicorn = bin_dir / "gunicorn"
+    if (
+        expected_gunicorn.exists()
+        and expected_gunicorn.is_file()
+        and os.access(expected_gunicorn, os.X_OK)
+    ):
+        # Found gunicorn in the same bin directory and it's executable
+        gunicorn_path = str(expected_gunicorn)
+        # Check if there's a different gunicorn in PATH
+        path_gunicorn = shutil.which("gunicorn")
+        warning_message = None
+        if path_gunicorn and Path(path_gunicorn).resolve() != expected_gunicorn:
+            warning_message = (
+                f"⚠️  WARNING: Multiple gunicorn installations detected!\n"
+                f"   Using: {gunicorn_path}\n"
+                f"   Found in PATH: {path_gunicorn}\n"
+                f"   This may cause version conflicts. Consider using a virtual environment."
+            )
+        return gunicorn_path, warning_message
+    # If file exists but isn't executable, add a warning about that
+    if expected_gunicorn.exists() and expected_gunicorn.is_file():
+        warning_message = (
+            f"⚠️  WARNING: gunicorn found at {expected_gunicorn} but it's not executable!\n"
+            f"   Falling back to PATH. Fix permissions with: chmod +x {expected_gunicorn}"
+        )
+        path_gunicorn = shutil.which("gunicorn")
+        if path_gunicorn:
+            return path_gunicorn, warning_message
+        # If not in PATH either, return error with permission hint
+        error_message = (
+            f"❌ ERROR: gunicorn found at {expected_gunicorn} but it's not executable!\n"
+            f"   Not found in PATH either.\n"
+            f"   Fix permissions with: chmod +x {expected_gunicorn}"
+        )
+        return "gunicorn", error_message
+    # Fall back to PATH
+    path_gunicorn = shutil.which("gunicorn")
+    if path_gunicorn:
+        warning_message = (
+            f"⚠️  WARNING: gunicorn not found in {bin_dir}\n"
+            f"   Falling back to gunicorn from PATH: {path_gunicorn}\n"
+            f"   This may cause issues if different versions are installed."
+        )
+        return path_gunicorn, warning_message
+    # Not found anywhere - return "gunicorn" and let subprocess.run fail with a clear error
+    warning_message = (
+        f"❌ ERROR: gunicorn not found!\n"
+        f"   Expected location: {expected_gunicorn}\n"
+    )
+    return "gunicorn", warning_message
 class PathResolver:
     """Centralized path resolution for both TT-Metal and upload/sync modes."""
@@ -143,6 +238,53 @@ def str_to_bool(string_value):
     return string_value.lower() in ("yes", "true", "t", "1")
+def is_running_in_container():
+    """
+    Detect if running inside a container (Docker, Podman, Kubernetes, etc.).
+    Uses multiple detection methods for robustness:
+    1. /.dockerenv file (Docker-specific, fastest check)
+    2. /proc/self/cgroup contains container indicators
+    3. Container-specific environment variables
+    Returns:
+        bool: True if running in a container, False otherwise
+    """
+    # Method 1: Check for /.dockerenv (Docker-specific, most common)
+    if os.path.exists("/.dockerenv"):
+        logger.info("Container detected via /.dockerenv file")
+        return True
+    # Method 2: Check cgroup for container indicators
+    try:
+        with open("/proc/self/cgroup", "r") as f:
+            content = f.read()
+            # Check for various container runtimes
+            container_indicators = ["docker", "containerd", "lxc", "kubepods"]
+            if any(indicator in content for indicator in container_indicators):
+                logger.info(
+                    f"Container detected via /proc/self/cgroup: {content[:100]}"
+                )
+                return True
+    except (FileNotFoundError, PermissionError):
+        # Not on Linux or no permission to read cgroup
+        pass
+    # Method 3: Check for container-specific environment variables
+    container_env_vars = [
+        "KUBERNETES_SERVICE_HOST",  # Kubernetes
+        "KUBERNETES_PORT",  # Kubernetes
+        "container",  # systemd-nspawn and others
+    ]
+    for env_var in container_env_vars:
+        if os.getenv(env_var):
+            logger.info(f"Container detected via environment variable: {env_var}")
+            return True
+    return False
 @dataclasses.dataclass
 class SerializeableDataclass:
     def to_dict(self) -> dict:

ttnn-visualizer 0.49.0__py3-none-any.whl → 0.64.0__py3-none-any.whl

ttnn-visualizer 0.49.0py3-none-any.whl → 0.64.0py3-none-any.whl