diff --git a/tests/analytical_visualization/app.py b/tests/analytical_visualization/app.py index b151e3c..f08b983 100644 --- a/tests/analytical_visualization/app.py +++ b/tests/analytical_visualization/app.py @@ -19,12 +19,18 @@ import streamlit as st # 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. +# Order matters: model_config first because everything else imports from it +# (e.g. TopologyConfig). If we reload downstream modules while model_config +# is stale, they end up holding the OLD dataclass definition and every +# `cfg.topo.new_field` lookup blows up with AttributeError. for _mod_name in ( + "tests.analytical_visualization.model_config", + "tests.analytical_visualization.model_presets", + "tests.analytical_visualization.autosuggest", + "tests.analytical_visualization.memory_layout", + "tests.analytical_visualization.stage_latencies", "tests.analytical_visualization.auto_explore", "tests.analytical_visualization.auto_hardware", - "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",