From da1909fdf2018ad5992b74ee96c1c9fc911d1bf3 Mon Sep 17 00:00:00 2001 From: Mukesh Garg Date: Tue, 28 Jul 2026 22:03:42 -0700 Subject: [PATCH] analytical-viz: extend hot-reload guard to all viz sub-modules Previously only reloaded auto_explore / auto_hardware / autosuggest / stage_latencies / memory_layout. When we edit pe_weight_layout, tensor_sharding, topology_map, pipeline_diagram, or optimization_report, Streamlit was still holding the pre-edit versions until a full restart. Add all five to the reload list so any visualization-module edit lands on the next Streamlit rerun without needing Ctrl+C. Co-Authored-By: Claude Opus 4.7 --- tests/analytical_visualization/app.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/analytical_visualization/app.py b/tests/analytical_visualization/app.py index 0c1da5d..7ae9a32 100644 --- a/tests/analytical_visualization/app.py +++ b/tests/analytical_visualization/app.py @@ -25,6 +25,11 @@ for _mod_name in ( "tests.analytical_visualization.autosuggest", "tests.analytical_visualization.stage_latencies", "tests.analytical_visualization.memory_layout", + "tests.analytical_visualization.pe_weight_layout", + "tests.analytical_visualization.tensor_sharding", + "tests.analytical_visualization.topology_map", + "tests.analytical_visualization.pipeline_diagram", + "tests.analytical_visualization.optimization_report", ): _m = sys.modules.get(_mod_name) if _m is not None: