refactor: index out of selectbox

dataforgoodfr · Apr 23, 2024 · bc7f1a5 · bc7f1a5
1 parent a4cab65
commit bc7f1a5
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/app/pages/4_Clean_Tables.py b/app/pages/4_Clean_Tables.py
@@ -136,16 +136,18 @@ def update_df_csv_to_save() -> None:
         )
 
     with col4:
+        index = (
+            list(st.session_state.tables.keys()).index(
+                st.session_state["algorithm_name"],
+            )
+            if "algorithm_name" in st.session_state
+            else 0
+        )
+
         st.session_state["algorithm_name"] = st.selectbox(
             "Choose the extracted table you want to see",
             list(st.session_state.tables.keys()),
-            index=(
-                list(st.session_state.tables.keys()).index(
-                    st.session_state["algorithm_name"],
-                )
-                if "algorithm_name" in st.session_state
-                else 0
-            ),
+            index=index,
             on_change=set_algorithm_name,
             args=("selectbox2",),
             key="selectbox2",