PyPI - ttnn-visualizer - Versions diffs - 0.40.3__py3-none-any.whl → 0.42.0__py3-none-any.whl - Mend

ttnn-visualizer 0.40.3py3-none-any.whl → 0.42.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 (23) hide show

ttnn_visualizer/static/assets/{splitPathsBySizeLoader-DoKJAUb8.js → splitPathsBySizeLoader-CauQGZHk.js} RENAMED Viewed

	@@ -1 +1 @@
1	- import{_ as o,a as _,b as i,p as c,I as u}from"./index-~~CWerbNbe~~.js";var p=function(n,s){return o(void 0,void 0,void 0,function(){var a,r;return _(this,function(e){switch(e.label){case 0:return a=c(n),s!==u.STANDARD?[3,2]:[4,i(()=>import("./index-BKzgFDAn.js").then(t=>t.I),[])];case 1:return r=e.sent(),[3,4];case 2:return[4,i(()=>import("./index-BvSuWPlB.js").then(t=>t.I),[])];case 3:r=e.sent(),e.label=4;case 4:return[2,r[a]]}})})};export{p as splitPathsBySizeLoader};
1	+ import{_ as o,a as _,b as i,p as c,I as u}from"./index-Ybr1HJxx.js";var p=function(n,s){return o(void 0,void 0,void 0,function(){var a,r;return _(this,function(e){switch(e.label){case 0:return a=c(n),s!==u.STANDARD?[3,2]:[4,i(()=>import("./index-BKzgFDAn.js").then(t=>t.I),[])];case 1:return r=e.sent(),[3,4];case 2:return[4,i(()=>import("./index-BvSuWPlB.js").then(t=>t.I),[])];case 3:r=e.sent(),e.label=4;case 4:return[2,r[a]]}})})};export{p as splitPathsBySizeLoader};

ttnn_visualizer/static/index.html CHANGED Viewed

@@ -13,17 +13,17 @@
         <link
             rel="icon"
             type="image/svg+xml"
-            href="/favicon/favicon.svg"
+            href="/static/favicon/favicon.svg"
         />
         <link
             rel="icon"
             type="image/png"
             sizes="32x32"
-            href="/favicon/favicon-32x32.png"
+            href="/static/favicon/favicon-32x32.png"
         />
         <link
             rel="manifest"
-            href="/assets/site-BTBrvHC5.webmanifest"
+            href="/static/assets/site-BTBrvHC5.webmanifest"
         />
         <meta
             name="theme-color"
@@ -34,8 +34,8 @@
             /* SERVER_CONFIG */
         </script>
-      <script type="module" crossorigin src="/assets/index-CWerbNbe.js"></script>
-      <link rel="stylesheet" crossorigin href="/assets/index-C1rJBrMl.css">
+      <script type="module" crossorigin src="/static/assets/index-Ybr1HJxx.js"></script>
+      <link rel="stylesheet" crossorigin href="/static/assets/index-C1rJBrMl.css">
     </head>
     <body>

ttnn_visualizer/tests/test_queries.py CHANGED Viewed

@@ -16,7 +16,6 @@ from ttnn_visualizer.models import (
 )
 from ttnn_visualizer.queries import DatabaseQueries
 from ttnn_visualizer.queries import LocalQueryRunner
-from ttnn_visualizer.queries import RemoteQueryRunner
 class TestQueryTable(unittest.TestCase):
@@ -222,14 +221,6 @@ class TestDatabaseQueries(unittest.TestCase):
             str(context.exception),
         )
-    @patch("ttnn_visualizer.queries.get_client")
-    def test_init_with_valid_remote_instance(self, _mock_client):
-        mock_instance = Mock()
-        mock_instance.remote_connection = Mock(useRemoteQuerying=True)
-        mock_instance.remote_connection.sqliteBinaryPath = "/usr/bin/sqlite3"
-        mock_instance.remote_profiler_folder = Mock(remotePath="/remote/path")
-        db_queries = DatabaseQueries(instance=mock_instance)
-        self.assertIsInstance(db_queries.query_runner, RemoteQueryRunner)
     def test_init_with_valid_local_instance(self):
         with tempfile.NamedTemporaryFile(suffix=".sqlite") as temp_db_file:
@@ -381,64 +372,5 @@ class TestDatabaseQueries(unittest.TestCase):
         self.assertEqual(result.operation_id, 2)
-class TestRemoteQueryRunner(unittest.TestCase):
-    def setUp(self):
-        self.mock_instance = Mock()
-        self.mock_instance.remote_connection.sqliteBinaryPath = "/usr/bin/sqlite3"
-        self.mock_instance.remote_connection.host = "mockhost"
-        self.mock_instance.remote_connection.user = "mockuser"
-        self.mock_instance.remote_profiler_folder.remotePath = "/remote/db"
-    @patch("ttnn_visualizer.queries.get_client")
-    def test_init_with_mock_get_client(self, mock_get_client):
-        # Mock the SSHClient returned by get_client
-        mock_ssh_client = Mock()
-        mock_get_client.return_value = mock_ssh_client
-        runner = RemoteQueryRunner(instance=self.mock_instance)
-        self.assertEqual(runner.ssh_client, mock_ssh_client)
-        mock_get_client.assert_called_once_with(
-            remote_connection=self.mock_instance.remote_connection
-        )
-    @patch("ttnn_visualizer.queries.get_client")
-    def test_execute_query(self, mock_get_client):
-        # Mock the SSH client
-        mock_ssh_client = Mock()
-        mock_get_client.return_value = mock_ssh_client
-        mock_stdout = Mock()
-        mock_stdout.read.return_value = b'[{"col1": "value1", "col2": "value2"}]'
-        mock_stderr = Mock()
-        mock_stderr.read.return_value = b""
-        mock_ssh_client.exec_command.return_value = (None, mock_stdout, mock_stderr)
-        runner = RemoteQueryRunner(instance=self.mock_instance)
-        query = "SELECT * FROM table WHERE id = ?"
-        params = [1]
-        results = runner.execute_query(query, params)
-        # Validate results
-        self.assertEqual(results, [("value1", "value2")])
-        mock_get_client.assert_called_once()
-        mock_ssh_client.exec_command.assert_called_once()
-    @patch("ttnn_visualizer.queries.get_client")
-    def test_close(self, mock_get_client):
-        # Mock the SSH client
-        mock_ssh_client = Mock()
-        mock_get_client.return_value = mock_ssh_client
-        runner = RemoteQueryRunner(instance=self.mock_instance)
-        runner.close()
-        mock_ssh_client.close.assert_called_once()
-    def tearDown(self):
-        pass
 if __name__ == "__main__":
     unittest.main()

ttnn_visualizer/views.py CHANGED Viewed

@@ -6,13 +6,13 @@ import dataclasses
 import json
 import logging
 import re
+import shutil
 import time
 from http import HTTPStatus
 from pathlib import Path
 from typing import List
-import shutil
-from wsgiref.validate import bad_header_value_re
+import yaml
 import zstd
 from flask import (
     Blueprint,
@@ -23,7 +23,8 @@ from flask import (
     request,
 )
-from ttnn_visualizer.csv_queries import DeviceLogProfilerQueries, OpsPerformanceQueries, OpsPerformanceReportQueries
+from ttnn_visualizer.csv_queries import DeviceLogProfilerQueries, OpsPerformanceQueries, OpsPerformanceReportQueries, \
+    NPEQueries
 from ttnn_visualizer.decorators import with_instance, local_only
 from ttnn_visualizer.enums import ConnectionTestStates
 from ttnn_visualizer.exceptions import DataFormatError
@@ -34,6 +35,9 @@ from ttnn_visualizer.file_uploads import (
     save_uploaded_files,
     validate_files,
 )
+from ttnn_visualizer.instances import (
+    get_instances, update_instance,
+)
 from ttnn_visualizer.models import (
     RemoteReportFolder,
     RemoteConnection,
@@ -51,9 +55,6 @@ from ttnn_visualizer.serializers import (
     serialize_operations_buffers,
     serialize_devices, serialize_buffer,
 )
-from ttnn_visualizer.instances import (
-    get_instances, update_instance,
-)
 from ttnn_visualizer.sftp_operations import (
     sync_remote_profiler_folders,
     read_remote_file,
@@ -64,14 +65,91 @@ from ttnn_visualizer.sftp_operations import (
     sync_remote_performance_folders,
     get_cluster_desc,
 )
-from ttnn_visualizer.ssh_client import get_client
+from ttnn_visualizer.exceptions import SSHException, AuthenticationException, NoValidConnectionsError
+import subprocess
 from ttnn_visualizer.utils import (
     get_cluster_descriptor_path,
     read_last_synced_file,
     timer,
 )
-import yaml
+def handle_ssh_subprocess_error(e: subprocess.CalledProcessError, remote_connection):
+    """
+    Convert subprocess SSH errors to appropriate SSH exceptions.
+    :param e: The subprocess.CalledProcessError
+    :param remote_connection: The RemoteConnection object for context
+    :raises: SSHException, AuthenticationException, or NoValidConnectionsError
+    """
+    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"
+    ]):
+        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"
+    ]):
+        raise NoValidConnectionsError(f"SSH connection failed: {e.stderr}")
+    # Check for general SSH protocol errors
+    elif "ssh:" in stderr or "protocol" in stderr:
+        raise SSHException(f"SSH protocol error: {e.stderr}")
+    # Default to generic SSH exception
+    else:
+        raise SSHException(f"SSH command failed: {e.stderr}")
+def test_ssh_connection(connection) -> bool:
+    """Test SSH connection by running a simple command."""
+    ssh_cmd = ["ssh"]
+    # Handle non-standard SSH port
+    if connection.port != 22:
+        ssh_cmd.extend(["-p", str(connection.port)])
+    ssh_cmd.extend([
+        f"{connection.username}@{connection.host}",
+        "echo 'SSH connection test'"
+    ])
+    try:
+        result = subprocess.run(
+            ssh_cmd,
+            capture_output=True,
+            text=True,
+            check=True,
+            timeout=10
+        )
+        return True
+    except subprocess.CalledProcessError as e:
+        if e.returncode == 255:  # SSH protocol errors
+            handle_ssh_subprocess_error(e, connection)
+        else:
+            raise RemoteConnectionException(
+                message=f"SSH connection test failed: {e.stderr}",
+                status=ConnectionTestStates.FAILED
+            )
+        return False
+    except subprocess.TimeoutExpired:
+        raise RemoteConnectionException(
+            message="SSH connection test timed out",
+            status=ConnectionTestStates.FAILED
+        )
+        return False
 logger = logging.getLogger(__name__)
@@ -636,6 +714,34 @@ def get_performance_data_raw(instance: Instance):
         headers={"Content-Disposition": "attachment; filename=profile_log_device.csv"},
     )
+@api.route("/performance/npe/manifest", methods=["GET"])
+@with_instance
+def get_npe_manifest(instance: Instance):
+    if not instance.performance_path:
+        return Response(status=HTTPStatus.NOT_FOUND)
+    try:
+        content = NPEQueries.get_npe_manifest(instance)
+    except FileNotFoundError:
+        return jsonify([])
+    return jsonify(content)
+@api.route("/performance/npe/timeline", methods=["GET"])
+@with_instance
+def get_npe_timeline(instance: Instance):
+    if not instance.performance_path:
+        return Response(status=HTTPStatus.NOT_FOUND)
+    filename = request.args.get("filename", default=None)
+    try:
+        content = NPEQueries.get_npe_timeline(instance, filename=filename)
+    except FileNotFoundError:
+        return jsonify({})
+    return jsonify(content)
 @api.route("/performance/device-log/zone/<zone>", methods=["GET"])
 @with_instance
@@ -901,7 +1007,7 @@ def test_remote_folder():
     # Test SSH Connection
     try:
-        get_client(connection)
+        test_ssh_connection(connection)
         add_status(ConnectionTestStates.OK.value, "SSH connection established")
     except RemoteConnectionException as e:
         add_status(ConnectionTestStates.FAILED.value, e.message)

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/LICENSE RENAMED Viewed

@@ -109,7 +109,6 @@ The following separate and independent dependencies are utilized by this project
 - Flask – BSD‑3‑Clause – https://github.com/pallets/flask/blob/main/LICENSE.txt
 - gunicorn – MIT – https://github.com/benoitc/gunicorn/blob/master/LICENSE
 - uvicorn – BSD‑3‑Clause – https://github.com/encode/uvicorn/blob/master/LICENSE.md
-- paramiko~=3.4.0 – LGPL‑2.1+ – https://github.com/paramiko/paramiko/blob/main/LICENSE
 - flask_cors – MIT – https://github.com/corydolphin/flask-cors/blob/main/LICENSE
 - pydantic – MIT – https://github.com/pydantic/pydantic-settings/blob/main/LICENSE
 - setuptools – MIT – https://github.com/pypa/setuptools/blob/main/LICENSE

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ttnn_visualizer
-Version: 0.40.3
+Version: 0.42.0
 Summary: TT-NN Visualizer
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -16,7 +16,6 @@ Requires-Dist: Flask-SocketIO==5.4.1
 Requires-Dist: Flask-SQLAlchemy==3.1.1
 Requires-Dist: pandas==2.2.3
 Requires-Dist: gunicorn~=22.0.0
-Requires-Dist: paramiko~=3.4.0
 Requires-Dist: pydantic==2.7.3
 Requires-Dist: pydantic-core==2.18.4
 Requires-Dist: setuptools==65.5.0
@@ -24,7 +23,7 @@ Requires-Dist: gevent===24.10.2
 Requires-Dist: python-dotenv==1.0.1
 Requires-Dist: sqlalchemy==2.0.34
 Requires-Dist: PyYAML==6.0.2
-Requires-Dist: tt-perf-report==1.0.6
+Requires-Dist: tt-perf-report==1.0.7
 Requires-Dist: zstd==1.5.7.0

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/RECORD RENAMED Viewed

@@ -1,46 +1,45 @@
 ttnn_visualizer/__init__.py,sha256=tVr9rTN1gXorz5XQQYmcRkzeEYjDGAKez1gP0oOIYeY,94
-ttnn_visualizer/app.py,sha256=bBVl93IhAncu5AOSeym2U7t8VIfe5TPQHUgkI9g5BTc,6770
-ttnn_visualizer/csv_queries.py,sha256=yFh3pUex6l9elGtSKhBVIZKEBOVmKyFpwTWozy1XxoY,22074
-ttnn_visualizer/decorators.py,sha256=YthndmmOfNlCMyjhChA_Lztcx1npFk7g0u4-7XT7Gnc,4571
+ttnn_visualizer/app.py,sha256=QCgd3nyc_I6iwJEAXGmNCvIXvR6oLSGKmLyV_7ECsH8,7222
+ttnn_visualizer/csv_queries.py,sha256=WzY1D_2fzrRzgdLnXf5w_aBwsW695Mkq5h2GXGeshyM,25676
+ttnn_visualizer/decorators.py,sha256=kveD2h6cgXD-rt03uAYWmZ9RKYphgvoGk53g2Pgn_RM,4799
 ttnn_visualizer/enums.py,sha256=SEIqp1tlc_zw2vQ8nHH9YTaV0m3Cb8fjn_goqz5wurE,203
-ttnn_visualizer/exceptions.py,sha256=iwDejaYUkRwHhpv8bc16X_2ax-2QmFqgaajbHvQImN0,1055
+ttnn_visualizer/exceptions.py,sha256=wzD_RuW1L6E2WdfM1pf34e25gisIe9jIMatf3HgyhIA,1375
 ttnn_visualizer/extensions.py,sha256=AW2aujcvayePfrwovVcsXIUkCFe_k2T4Vk3OblZd4Bk,380
 ttnn_visualizer/file_uploads.py,sha256=aXolaGeisY3H89y_GxfciPvkSU1XmcvWqpf8vDqKrgM,5011
 ttnn_visualizer/instances.py,sha256=99L80yT0BXyhEkBbGFC_tNAg8LlGWUqWQrhkK-4-Yxc,11261
 ttnn_visualizer/models.py,sha256=ut9V9fTzbbjk_WNKKP8ILrVXYnIv7qQaDvNOEEX3BT8,7778
-ttnn_visualizer/queries.py,sha256=KC4WtI6Pf3LL_LqHXQTC5zVd-HvjJLWY4Stndd0z0qs,13681
-ttnn_visualizer/remote_sqlite_setup.py,sha256=FXZTT4_-7zPG6rooXJ7o3a12r86hRzKt87qfUiKdMcs,3269
-ttnn_visualizer/requirements.txt,sha256=AoRskxD-FUPk1j_xruPxweDgKb3RjFv8lCQNqpf6Rrs,370
+ttnn_visualizer/queries.py,sha256=MB57Ci3gyXn9jdXdR6KaMBHoHYt5jgdk3wP7jSDGWFA,9891
+ttnn_visualizer/remote_sqlite_setup.py,sha256=0JfSQLA_ST6gh7lyHhxUdsEsmw8moremC37myES6Al8,5892
+ttnn_visualizer/requirements.txt,sha256=RuHRxxkHKSAAMF7JbvifDMOOFTfQH19AW3qJFL3I69M,339
 ttnn_visualizer/serializers.py,sha256=7kjtxp3NEpTlBKUNsxdtUcOM23pAyIt5MR-XTl85bVw,8028
 ttnn_visualizer/settings.py,sha256=r7UdqFuXEVg0qw1F2_zRFM_687PctqpgoUEKjvTVhwU,4578
-ttnn_visualizer/sftp_operations.py,sha256=Sn13vkkQoRifCO3SCNGo6ORlwaHB4MO4uSxxo_7m8Dk,17906
+ttnn_visualizer/sftp_operations.py,sha256=p06_dYHoM-qbAZuUUR9CN7fWy4U_QpIaXBxPsjxJfgI,32074
 ttnn_visualizer/sockets.py,sha256=GAvReqKaT0YQyeb0kf7ndtL1odLldHCpbbKkr0INXRM,3624
-ttnn_visualizer/ssh_client.py,sha256=dTkxzAFCGKyiZL1_LHx23Gn4y0oW4yKfGKjTxuqcx8I,2679
 ttnn_visualizer/utils.py,sha256=9Ier5XBD4vHzvtIxfbIRnsP1Is03J2usUmhfB5gSqqc,6186
-ttnn_visualizer/views.py,sha256=MlbN19BYaVgsVxYwN0pqvqqRmOsDPy9gFGjTZosyUUI,41465
+ttnn_visualizer/views.py,sha256=WywffIzF5ahslw6qBDHyfL5kIzDKxbUf_yQXxeMUVgA,44784
 ttnn_visualizer/bin/docker-entrypoint-web,sha256=uuv6aubpMCfOcuvDBxwBDITE8PN39teuwyJ2zA5KWuw,413
 ttnn_visualizer/bin/pip3-install,sha256=nbSRT4GfJQIQ9KTNO3j-6b5WM4lrx9XA4GBlAURRMws,502
-ttnn_visualizer/static/index.html,sha256=kFTLNN94zVVgKgkuZ6NKCQnVNraDQ6Hrjsyru1JwHT0,1100
-ttnn_visualizer/static/assets/allPaths-DSFl5HNA.js,sha256=4rwRYkqQBalBP6PUs5qtVWI0BsBTuTcc7r8_Y7SoP30,309
-ttnn_visualizer/static/assets/allPathsLoader-BwkPDbOI.js,sha256=zS7B8U-iYYt6SiEhejfFcjjgWT-fm8OL13TOy9VnFSs,550
+ttnn_visualizer/static/index.html,sha256=gM7cQnEuPT29ZMKqJcEqjW9POf4AigZftNS8uw68ViY,1135
+ttnn_visualizer/static/assets/allPaths-wwXsGKJ2.js,sha256=CxiByiF9vD2BvhrONDu-MQVoLKj-KhOcYm1XBZdSXAw,309
+ttnn_visualizer/static/assets/allPathsLoader-BK9jqlVe.js,sha256=Bc4W_X5zso_Yf-8g1aZBA9wg0NfKXGjD64lDJWqAyzE,550
 ttnn_visualizer/static/assets/index-BKzgFDAn.js,sha256=Pu504I3PJMoop2k0KQxrd1qKuS3zR5KtvTTdz9IhfeY,283793
 ttnn_visualizer/static/assets/index-BvSuWPlB.js,sha256=43E6se9tLaZkX7kCG4waDG_vyL4yED6SEaDKLt3Nw_s,292382
 ttnn_visualizer/static/assets/index-C1rJBrMl.css,sha256=f1PX54ZLwJIs0PqePOu-B3YP3joUxhrkkZ1Butqil9M,623666
-ttnn_visualizer/static/assets/index-CWerbNbe.js,sha256=yI0Gdggx7pxJ9T4eGZfTnSnFTwjThYqHKIBEhu6pCJU,7784873
+ttnn_visualizer/static/assets/index-Ybr1HJxx.js,sha256=fSByVEefEo8y-kGOi8sS1n6_thxMXcapSWptHAd3z3Y,7785792
 ttnn_visualizer/static/assets/site-BTBrvHC5.webmanifest,sha256=Uy_XmnGuYFVf-OZuma2NvgEPdrCrevb3HZvaxSIHoA0,456
-ttnn_visualizer/static/assets/splitPathsBySizeLoader-DoKJAUb8.js,sha256=O8apTiCHTzNdudN5X__rXR7br9w6mlPREJWAcp7Hl-g,472
+ttnn_visualizer/static/assets/splitPathsBySizeLoader-CauQGZHk.js,sha256=431X0PrDog4lr5A_hi5ObsXMZe6KpjgIDgTc65IwZWM,472
 ttnn_visualizer/static/favicon/android-chrome-192x192.png,sha256=BZWA09Zxaa3fXbaeS6nhWo2e-DUSjm9ElzNQ_xTB5XU,6220
 ttnn_visualizer/static/favicon/android-chrome-512x512.png,sha256=HBiJSZyguB3o8fMJuqIGcpeBy_9JOdImme3wD02UYCw,62626
 ttnn_visualizer/static/favicon/favicon-32x32.png,sha256=Zw201qUsczQv1UvoQvJf5smQ2ss10xaTeWxmQNYCGtY,480
 ttnn_visualizer/static/favicon/favicon.svg,sha256=wDPY3VrekJ_DE1TnJ2vUy602K3S4Xe9TgrdZ7jXK9c8,633
 ttnn_visualizer/static/sample-data/cluster-desc.yaml,sha256=LMxOmsRUXtVVU5ogzYkXUozB3dg2IzqIRJQpV_O5qMU,29618
 ttnn_visualizer/tests/__init__.py,sha256=tVr9rTN1gXorz5XQQYmcRkzeEYjDGAKez1gP0oOIYeY,94
-ttnn_visualizer/tests/test_queries.py,sha256=74JgzuA3Kj1DPjLexI-YCOl8sgMi9L1I1wkNdzeHih8,16602
+ttnn_visualizer/tests/test_queries.py,sha256=QLfV_MWnM68WOWKKa1TPzn5aKQJbysQrGPIJ5aoxl7k,13882
 ttnn_visualizer/tests/test_serializers.py,sha256=XyS5f8JK234CtizslrQ49hhxaz7RmhkEbVTaxOvdSd4,18490
-ttnn_visualizer-0.40.3.dist-info/LICENSE,sha256=hcHiC16In8fLrs56SK2KzaBvs1OwBfVMJBxi7Rc2w0g,14899
-ttnn_visualizer-0.40.3.dist-info/LICENSE_understanding.txt,sha256=pymi-yb_RvYM9p2ZA4iSNsImcvhDBBxlGuJCY9dTq7M,233
-ttnn_visualizer-0.40.3.dist-info/METADATA,sha256=T70sbFH3NeezG2q6DmaRVX0zUn-myPAy7HIBhY0IwGY,7393
-ttnn_visualizer-0.40.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ttnn_visualizer-0.40.3.dist-info/entry_points.txt,sha256=QpuUpkmQ_mEHJTMqOBdU0MH2Z4WF_9iFsGACeyyAO1E,61
-ttnn_visualizer-0.40.3.dist-info/top_level.txt,sha256=M1EGkvDOuIfbhDbcUdz2-TSdmCtDoQ2Uyag9k5JLDSY,16
-ttnn_visualizer-0.40.3.dist-info/RECORD,,
+ttnn_visualizer-0.42.0.dist-info/LICENSE,sha256=-Y7CZK1-MxZZ7l0Rb414S0SZ9tdCby-bMGl_LmFFicM,14806
+ttnn_visualizer-0.42.0.dist-info/LICENSE_understanding.txt,sha256=pymi-yb_RvYM9p2ZA4iSNsImcvhDBBxlGuJCY9dTq7M,233
+ttnn_visualizer-0.42.0.dist-info/METADATA,sha256=PpiZyDS_Vxcen6rGwhwgx1YFrfmCb35x7Jn_mxijYgU,7362
+ttnn_visualizer-0.42.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ttnn_visualizer-0.42.0.dist-info/entry_points.txt,sha256=QpuUpkmQ_mEHJTMqOBdU0MH2Z4WF_9iFsGACeyyAO1E,61
+ttnn_visualizer-0.42.0.dist-info/top_level.txt,sha256=M1EGkvDOuIfbhDbcUdz2-TSdmCtDoQ2Uyag9k5JLDSY,16
+ttnn_visualizer-0.42.0.dist-info/RECORD,,

ttnn_visualizer/ssh_client.py DELETED Viewed

@@ -1,85 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0
-#
-# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
-import paramiko
-import os
-from pathlib import Path
-from paramiko.agent import Agent
-from paramiko.ssh_exception import SSHException
-from ttnn_visualizer.decorators import remote_exception_handler
-from ttnn_visualizer.models import RemoteConnection
-import logging
-logger = logging.getLogger(__name__)
-def initialize_ssh_client():
-    ssh = paramiko.SSHClient()
-    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-    ssh.load_system_host_keys()
-    return ssh
-def get_connection_args(remote_connection: RemoteConnection) -> dict:
-    use_agent = os.getenv("USE_SSH_AGENT", "true").lower() == "true"
-    ssh_config_path = Path(os.getenv("SSH_CONFIG_PATH", "~/.ssh/config")).expanduser()
-    if use_agent:
-        agent = Agent()
-        keys = agent.get_keys()
-        if not keys:
-            logger.error("No keys found in agent")
-            raise SSHException("No keys found")
-        return {"look_for_keys": True}
-    config = paramiko.SSHConfig.from_path(ssh_config_path).lookup(
-        remote_connection.host
-    )
-    if not config:
-        raise SSHException(f"Host not found in SSH config {remote_connection.host}")
-    return {"key_filename": config["identityfile"].pop(), "look_for_keys": False}  # type: ignore
-@remote_exception_handler
-def get_client(remote_connection: RemoteConnection) -> paramiko.SSHClient:
-    ssh = initialize_ssh_client()
-    connection_args = get_connection_args(remote_connection)
-    ssh.connect(
-        remote_connection.host,
-        port=remote_connection.port,
-        username=remote_connection.username,
-        **connection_args,
-    )
-    return ssh
-def check_permissions(client, directory):
-    """Check if write and delete permissions are available in the remote directory."""
-    test_file = Path(directory) / "test_permission_file.txt"
-    touch_command = f"touch {test_file}"
-    remove_command = f"rm {test_file}"
-    stdin, stdout, stderr = client.exec_command(touch_command)
-    error = stderr.read().decode().strip()
-    if error:
-        raise Exception(f"No permission to create files in {directory}: {error}")
-    stdin, stdout, stderr = client.exec_command(remove_command)
-    error = stderr.read().decode().strip()
-    if error:
-        raise Exception(f"No permission to delete files in {directory}: {error}")
-    return True
-def check_gzip_exists(client):
-    """Check if gzip and tar exist on the remote server."""
-    stdin, stdout, stderr = client.exec_command("which gzip && which tar")
-    result = stdout.read().decode().strip()
-    if not result:
-        return False
-    return True

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/LICENSE_understanding.txt RENAMED Viewed

File without changes

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/WHEEL RENAMED Viewed

File without changes

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/entry_points.txt RENAMED Viewed

File without changes

{ttnn_visualizer-0.40.3.dist-info → ttnn_visualizer-0.42.0.dist-info}/top_level.txt RENAMED Viewed

File without changes

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

ttnn-visualizer 0.40.3py3-none-any.whl → 0.42.0py3-none-any.whl