ttnn-visualizer 0.36.0__py3-none-any.whl → 0.37.1__py3-none-any.whl

ttnn_visualizer/queries.py

@@ -31,20 +31,20 @@ import paramiko
 
 
 class LocalQueryRunner:
-    def __init__(self, session: Optional[Instance] = None, connection=None):
+    def __init__(self, instance: Optional[Instance] = None, connection=None):
 
         if connection:
             self.connection = connection
         else:
-            if not session or not session.profiler_path:
+            if not instance or not instance.profiler_path:
                 raise ValueError("Report path must be provided for local queries")
-            db_path = str(session.profiler_path)
+            db_path = str(instance.profiler_path)
             if not Path(db_path).exists():
                 raise DatabaseFileNotFoundException(
                     f"Database not found at path: {db_path}"
                 )
             self.connection = sqlite3.connect(
-                session.profiler_path, isolation_level=None, timeout=30
+                instance.profiler_path, isolation_level=None, timeout=30
             )
 
     def execute_query(self, query: str, params: Optional[List] = None) -> List:
@@ -66,24 +66,24 @@ class LocalQueryRunner:
 class RemoteQueryRunner:
     column_delimiter = "|||"
 
-    def __init__(self, session: Instance):
-        self.session = session
-        self._validate_session()
-        self.ssh_client = self._get_ssh_client(self.session.remote_connection)
-        self.sqlite_binary = self.session.remote_connection.sqliteBinaryPath
+    def __init__(self, instance: Instance):
+        self.instance = instance
+        self._validate_instance()
+        self.ssh_client = self._get_ssh_client(self.instance.remote_connection)
+        self.sqlite_binary = self.instance.remote_connection.sqliteBinaryPath
         self.remote_db_path = str(
-            Path(self.session.remote_profiler_folder.remotePath, "db.sqlite")
+            Path(self.instance.remote_profiler_folder.remotePath, "db.sqlite")
         )
 
-    def _validate_session(self):
+    def _validate_instance(self):
         """
-        Validate that the session has all required remote connection attributes.
+        Validate that the instance has all required remote connection attributes.
         """
         if (
-            not self.session.remote_connection
-            or not self.session.remote_connection.sqliteBinaryPath
-            or not self.session.remote_profiler_folder
-            or not self.session.remote_profiler_folder.remotePath
+            not self.instance.remote_connection
+            or not self.instance.remote_connection.sqliteBinaryPath
+            or not self.instance.remote_profiler_folder
+            or not self.instance.remote_profiler_folder.remotePath
         ):
             raise ValueError(
                 "Remote connections require remote path and sqliteBinaryPath"
@@ -141,9 +141,9 @@ class RemoteQueryRunner:
 
     def execute_query(self, query: str, params: Optional[List] = None) -> List:
         """
-        Execute a remote SQLite query using the session's SSH client.
+        Execute a remote SQLite query using the instance's SSH client.
         """
-        self._validate_session()
+        self._validate_instance()
         formatted_query = self._format_query(query, params)
         command = self._build_command(formatted_query)
         output, error_output = self._execute_ssh_command(command)
@@ -165,25 +165,25 @@ class RemoteQueryRunner:
 
 class DatabaseQueries:
 
-    session: Optional[Instance] = None
+    instance: Optional[Instance] = None
     ssh_client = None
     query_runner: LocalQueryRunner | RemoteQueryRunner
 
-    def __init__(self, session: Optional[Instance] = None, connection=None):
-        self.session = session
+    def __init__(self, instance: Optional[Instance] = None, connection=None):
+        self.instance = instance
 
         if connection:
             self.query_runner = LocalQueryRunner(connection=connection)
         else:
-            if not session:
+            if not instance:
                 raise ValueError(
-                    "Must provide either an existing connection or session"
+                    "Must provide either an existing connection or instance"
                 )
-            remote_connection = session.remote_connection if session else None
+            remote_connection = instance.remote_connection if instance else None
             if remote_connection and remote_connection.useRemoteQuerying:
-                self.query_runner = RemoteQueryRunner(session=session)
+                self.query_runner = RemoteQueryRunner(instance=instance)
             else:
-                self.query_runner = LocalQueryRunner(session=session)
+                self.query_runner = LocalQueryRunner(instance=instance)
 
     def _check_table_exists(self, table_name: str) -> bool:
         """

ttnn_visualizer/serializers.py

@@ -1,15 +1,13 @@
+# SPDX-License-Identifier: Apache-2.0
+#
+# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+
 import dataclasses
 from collections import defaultdict
 from typing import List
 
 from ttnn_visualizer.models import BufferType, Operation, TensorComparisonRecord
 
-
-# SPDX-License-Identifier: Apache-2.0
-#
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
-
-
 def serialize_operations(
     inputs,
     operation_arguments,

ttnn_visualizer/settings.py

@@ -16,16 +16,23 @@ class DefaultConfig(object):
     DEBUG = bool(str_to_bool(os.getenv("FLASK_DEBUG", "false")))
     TESTING = False
     PRINT_ENV = True
+    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(",")
+        if o
+    ]
 
     # Path Settings
     DB_VERSION = "0.29.0"  # App version when DB schema last changed
-    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"
     PERFORMANCE_DIRECTORY_NAME = "performance-reports"
     NPE_DIRECTORY_NAME = "npe-reports"
     APPLICATION_DIR = os.path.abspath(os.path.join(__file__, "..", os.pardir))
+    APP_DATA_DIRECTORY = os.getenv("APP_DATA_DIRECTORY", APPLICATION_DIR)
     STATIC_ASSETS_DIR = Path(APPLICATION_DIR).joinpath("ttnn_visualizer", "static")
     SEND_FILE_MAX_AGE_DEFAULT = 0
 
@@ -40,7 +47,7 @@ class DefaultConfig(object):
 
     # SQL Alchemy Settings
     SQLALCHEMY_DATABASE_URI = (
-        f"sqlite:///{os.path.join(APPLICATION_DIR, f'ttnn_{DB_VERSION}.db')}"
+        f"sqlite:///{os.path.join(APP_DATA_DIRECTORY, f'ttnn_{DB_VERSION}.db')}"
     )
     SQLALCHEMY_ENGINE_OPTIONS = {
         "pool_size": 10,  # Adjust pool size as needed (default is 5)

ttnn_visualizer/sftp_operations.py

@@ -52,7 +52,7 @@ def resolve_file_path(remote_connection, file_path: str) -> str:
     """
     Resolve the file path if it contains a wildcard ('*') by using glob on the remote machine.
 
-    :param session: A session object containing the remote connection information.
+    :param remote_connection: A RemoteConnection object containing the remote connection information.
     :param file_path: The file path, which may include wildcards.
     :return: The resolved file path.
     :raises FileNotFoundError: If no files match the pattern.
@@ -286,9 +286,13 @@ def get_remote_profiler_folder_from_config_path(
     attributes = sftp.lstat(str(config_path))
     with sftp.open(str(config_path), "rb") as config_file:
         data = json.loads(config_file.read())
+
+        report_name = data.get("report_name")
+        logger.info(f"********* report_name: {report_name}")
+
         return RemoteReportFolder(
             remotePath=str(Path(config_path).parent),
-            testName=data["report_name"], # Config file includes "report_name"
+            reportName=report_name,
             lastModified=(
                 int(attributes.st_mtime) if attributes.st_mtime else int(time.time())
             ),
@@ -307,7 +311,7 @@ def get_remote_performance_folder(
     )
     return RemoteReportFolder(
         remotePath=str(remote_path),
-        testName=str(performance_name),
+        reportName=str(performance_name),
         lastModified=last_modified,
     )
 

ttnn_visualizer/static/assets/{allPaths-ChIeDZ5t.js → allPaths-BaKDx9R3.js}

@@ -1 +1 @@
-import{I as n}from"./index-BKzgFDAn.js";import{I as e}from"./index-BvSuWPlB.js";import{p as r,I as s}from"./index-D8zG3DIo.js";function I(o,t){var a=r(o);return t===s.STANDARD?n[a]:e[a]}function p(o){return r(o)}export{n as IconSvgPaths16,e as IconSvgPaths20,I as getIconPaths,p as iconNameToPathsRecordKey};
+import{I as n}from"./index-BKzgFDAn.js";import{I as e}from"./index-BvSuWPlB.js";import{p as r,I as s}from"./index-wX7GhuKE.js";function I(o,t){var a=r(o);return t===s.STANDARD?n[a]:e[a]}function p(o){return r(o)}export{n as IconSvgPaths16,e as IconSvgPaths20,I as getIconPaths,p as iconNameToPathsRecordKey};

ttnn_visualizer/static/assets/{allPathsLoader-C4OHN8TU.js → allPathsLoader-DVskN3_V.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-ChIeDZ5t.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-D8zG3DIo.js","assets/index-BnUuxY3c.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-D8zG3DIo.js";var _=function(e,a){return o(void 0,void 0,void 0,function(){var t;return n(this,function(r){switch(r.label){case 0:return[4,i(()=>import("./allPaths-ChIeDZ5t.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-BaKDx9R3.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-wX7GhuKE.js","assets/index-je2tF5Bg.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-wX7GhuKE.js";var _=function(e,a){return o(void 0,void 0,void 0,function(){var t;return n(this,function(r){switch(r.label){case 0:return[4,i(()=>import("./allPaths-BaKDx9R3.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};