ttnn-visualizer 0.42.0__py3-none-any.whl → 0.43.1__py3-none-any.whl

ttnn_visualizer/remote_sqlite_setup.py

@@ -7,11 +7,18 @@ import subprocess
 
 from ttnn_visualizer.decorators import remote_exception_handler
 from ttnn_visualizer.enums import ConnectionTestStates
-from ttnn_visualizer.exceptions import RemoteSqliteException, SSHException, AuthenticationException, NoValidConnectionsError
+from ttnn_visualizer.exceptions import (
+    RemoteSqliteException,
+    SSHException,
+    AuthenticationException,
+    NoValidConnectionsError,
+)
 from ttnn_visualizer.models import RemoteConnection
 
 
-def handle_ssh_subprocess_error(e: subprocess.CalledProcessError, remote_connection: RemoteConnection):
+def handle_ssh_subprocess_error(
+    e: subprocess.CalledProcessError, remote_connection: RemoteConnection
+):
     """
     Convert subprocess SSH errors to appropriate SSH exceptions.
 
@@ -22,23 +29,29 @@ def handle_ssh_subprocess_error(e: subprocess.CalledProcessError, remote_connect
     stderr = e.stderr.lower() if e.stderr else ""
 
     # Check for authentication failures
-    if any(auth_err in stderr for auth_err in [
-        "permission denied",
-        "authentication failed",
-        "publickey",
-        "password",
-        "host key verification failed"
-    ]):
+    if any(
+        auth_err in stderr
+        for auth_err in [
+            "permission denied",
+            "authentication failed",
+            "publickey",
+            "password",
+            "host key verification failed",
+        ]
+    ):
         raise AuthenticationException(f"SSH authentication failed: {e.stderr}")
 
     # Check for connection failures
-    elif any(conn_err in stderr for conn_err in [
-        "connection refused",
-        "network is unreachable",
-        "no route to host",
-        "name or service not known",
-        "connection timed out"
-    ]):
+    elif any(
+        conn_err in stderr
+        for conn_err in [
+            "connection refused",
+            "network is unreachable",
+            "no route to host",
+            "name or service not known",
+            "connection timed out",
+        ]
+    ):
         raise NoValidConnectionsError(f"SSH connection failed: {e.stderr}")
 
     # Check for general SSH protocol errors
@@ -49,29 +62,23 @@ def handle_ssh_subprocess_error(e: subprocess.CalledProcessError, remote_connect
     else:
         raise SSHException(f"SSH command failed: {e.stderr}")
 
+
 MINIMUM_SQLITE_VERSION = "3.38.0"
 
 
 def _execute_ssh_command(remote_connection: RemoteConnection, command: str) -> str:
     """Execute an SSH command and return the output."""
-    ssh_cmd = ["ssh"]
-    
+    ssh_cmd = ["ssh", "-o", "PasswordAuthentication=no"]
+
     # Handle non-standard SSH port
     if remote_connection.port != 22:
         ssh_cmd.extend(["-p", str(remote_connection.port)])
-    
-    ssh_cmd.extend([
-        f"{remote_connection.username}@{remote_connection.host}",
-        command
-    ])
-    
+
+    ssh_cmd.extend([f"{remote_connection.username}@{remote_connection.host}", command])
+
     try:
         result = subprocess.run(
-            ssh_cmd,
-            capture_output=True,
-            text=True,
-            check=True,
-            timeout=30
+            ssh_cmd, capture_output=True, text=True, check=True, timeout=30
         )
         return result.stdout
     except subprocess.CalledProcessError as e:
@@ -118,7 +125,7 @@ def is_sqlite_executable(remote_connection: RemoteConnection, binary_path):
     try:
         output = _execute_ssh_command(remote_connection, f"{binary_path} --version")
         version_output = output.strip()
-        
+
         version = get_sqlite_version(version_output)
         if not is_version_at_least(version, MINIMUM_SQLITE_VERSION):
             raise Exception(

ttnn_visualizer/requirements.txt

@@ -20,3 +20,4 @@ zstd==1.5.7.0
 
 # Dev dependencies
 mypy
+black==25.1.0

ttnn_visualizer/serializers.py

@@ -8,6 +8,7 @@ from typing import List
 
 from ttnn_visualizer.models import BufferType, Operation, TensorComparisonRecord
 
+
 def serialize_operations(
     inputs,
     operation_arguments,

ttnn_visualizer/settings.py

@@ -10,6 +10,7 @@ from ttnn_visualizer.utils import str_to_bool
 
 load_dotenv()
 
+
 class DefaultConfig(object):
     # General Settings
     SECRET_KEY = os.getenv("SECRET_KEY", "90909")
@@ -19,17 +20,20 @@ class DefaultConfig(object):
     SERVER_MODE = str_to_bool(os.getenv("SERVER_MODE", "false"))
     MALWARE_SCANNER = os.getenv("MALWARE_SCANNER")
     ALLOWED_ORIGINS = [
-        o for o in os.getenv("ALLOWED_ORIGINS", "http://localhost:5173,http://localhost:8000").split(",")
+        o
+        for o in os.getenv(
+            "ALLOWED_ORIGINS", "http://localhost:5173,http://localhost:8000"
+        ).split(",")
         if o
     ]
     BASE_PATH = os.getenv("BASE_PATH", "/")
-    MAX_CONTENT_LENGTH = (
-        None if not (v := os.getenv("MAX_CONTENT_LENGTH")) else int(v)
-    )
+    MAX_CONTENT_LENGTH = None if not (v := os.getenv("MAX_CONTENT_LENGTH")) else int(v)
 
     # Path Settings
     DB_VERSION = "0.29.0"  # App version when DB schema last changed
-    REPORT_DATA_DIRECTORY = os.getenv("REPORT_DATA_DIRECTORY", Path(__file__).parent.absolute().joinpath("data"))
+    REPORT_DATA_DIRECTORY = os.getenv(
+        "REPORT_DATA_DIRECTORY", Path(__file__).parent.absolute().joinpath("data")
+    )
     LOCAL_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("local")
     REMOTE_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("remote")
     PROFILER_DIRECTORY_NAME = "profiler-reports"