Bug fix

rte-france · Dec 19, 2024 · 96d393d · 96d393d
1 parent 10cf9a5
commit 96d393d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bertrend/trend_analysis/visualizations.py b/bertrend/trend_analysis/visualizations.py
@@ -146,7 +146,7 @@ def plot_topics_per_timestamp(topic_models: Dict[pd.Timestamp, BERTopic]) -> Non
 def create_topic_size_evolution_figure(topic_ids=None):
     fig = go.Figure()
 
-    topic_sizes = st.session_state.topic_sizes
+    topic_sizes = st.session_state["bertrend"].topic_sizes
 
     if topic_ids is None:
         # If topic_ids is not provided, include all topics
@@ -201,9 +201,9 @@ def plot_topic_size_evolution(
         Tuple[float, float]: The q1 and q3 values representing the 10th and 90th percentiles of popularity values.
     """
 
-    topic_sizes = st.session_state.topic_sizes
-    topic_last_popularity = st.session_state.topic_last_popularity
-    topic_last_update = st.session_state.topic_last_update
+    topic_sizes = st.session_state["bertrend"].topic_sizes
+    topic_last_popularity = st.session_state["bertrend"].topic_last_popularity
+    topic_last_update = st.session_state["bertrend"].topic_last_update
 
     window_size_timedelta = pd.Timedelta(days=window_size)
     granularity_timedelta = pd.Timedelta(days=granularity)