diff --git a/tests/analytical_visualization/app.py b/tests/analytical_visualization/app.py
index 9ab82df..bf4e97b 100644
--- a/tests/analytical_visualization/app.py
+++ b/tests/analytical_visualization/app.py
@@ -6,10 +6,30 @@ Run:
 """
 from __future__ import annotations
 
+import importlib
+import sys
+
 import matplotlib.pyplot as plt
 import pandas as pd
 import streamlit as st
 
+# Streamlit's hot-reload only re-runs THIS script; imported sub-modules stay
+# cached in sys.modules across reruns. When we edit auto_explore.py or
+# auto_hardware.py while the Streamlit process is alive, the app would
+# keep using the pre-edit versions until a full Ctrl+C + restart. Force
+# a reload of our own modules on every rerun so signature changes land
+# without a full restart.
+for _mod_name in (
+    "tests.analytical_visualization.auto_explore",
+    "tests.analytical_visualization.auto_hardware",
+    "tests.analytical_visualization.autosuggest",
+    "tests.analytical_visualization.stage_latencies",
+    "tests.analytical_visualization.memory_layout",
+):
+    _m = sys.modules.get(_mod_name)
+    if _m is not None:
+        importlib.reload(_m)
+
 from tests.analytical_visualization.model_config import (
     FullConfig, ModelConfig, TopologyConfig, MachineParams,
 )