Bug fix

rte-france · Dec 23, 2024 · 90a764a · 90a764a
1 parent 4b44aca
commit 90a764a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/bertrend/demos/weak_signals/app.py b/bertrend/demos/weak_signals/app.py
@@ -211,6 +211,12 @@ def main():
     with tab2:
         st.header("Model Training")
 
+        if not SessionStateManager.get("data_embedded", False):
+            st.warning(
+                "Please embed data and train models before proceeding to analysis."
+            )
+            st.stop()
+
         # Select granularity
         st.number_input(
             "Select Granularity",