Use updated constants in runner

jvivian · Dec 7, 2023 · a2cdf42 · a2cdf42
1 parent 6c0f4f7
commit a2cdf42
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/covid19_drdfm/streamlit/runner.py b/covid19_drdfm/streamlit/runner.py
@@ -12,6 +12,7 @@
 
 from covid19_drdfm.dfm import state_process
 from covid19_drdfm.processing import get_df, get_factors
+from covid19_drdfm.processing import NAME_MAP
 
 st.set_page_config(layout="wide")
 pio.templates.default = "plotly_white"
@@ -31,6 +32,8 @@
     "Inflation": ["Monetary_2", "Monetary_3", "Monetary_1"],
     "Pandemic": ["Pandemic_1", "Pandemic_2", "Pandemic_6", "Pandemic_9", "Pandemic_7", "Pandemic_10"],
 }
+DEFAULTS = {NAME_MAP[x]: [NAME_MAP[z] for z in y] for x, y in DEFAULTS.items() if x in NAME_MAP in NAME_MAP}
+print(DEFAULTS)
 
 
 def center_title(text):