Update my_pages/information_loss.py

my_pages/information_loss.py

@@ -1,6 +1,5 @@
 import pandas as pd
 import streamlit as st
-from streamlit_extras.stylable_container import stylable_container
 from utils import go_to
 
 ALL_FEATURES = [
@@ -20,7 +19,7 @@ def render():
     st.markdown(
         """
         <style>
-        button[kind="selected"] {
+        button[kind="primary"] {
             background: green!important;
         }
         </style>
@@ -63,14 +62,7 @@ def render():
         for feature in ALL_FEATURES.copy():
             if feature in st.session_state.selected_features:
                 feature_name = feature[1]
-                with stylable_container(
-                    feature_name,
-                    css_styles=f"""
-                    button {{
-                        background: green!important;            
-                    }}""",
-                ):
-                    _ = st.button(feature_name, type="selected")
+                _ = st.button(feature_name, type="primary")
             else:
                 feature_name = feature[0]
                 if st.button(feature_name):