viz

app.py

@@ -85,7 +85,7 @@ def data_comparison(df):
         y=alt.Y('slice:N', axis=alt.Axis(orient='left'), title=""),
         x=alt.X("label"),
         shape=alt.Shape('label', scale=alt.Scale(
-            range=['circle', 'diamond']), legend=['neg', 'pos']),
+            range=['circle', 'diamond']), legend=None),
         color=color
     ).add_selection(
         selection
@@ -214,6 +214,7 @@ if __name__ == "__main__":
     )
     ### LOAD DATA AND SESSION VARIABLES ###
     data_df = pd.read_parquet('./assets/data/amazon_polarity.test.parquet')
+    data_df.reset_index(drop=True, inplace=True)
     embedding_umap = data_df[['x','y']]
     if "user_data" not in st.session_state:
         st.session_state["user_data"] = data_df

error_analysis/utils/style_hacks.py

@@ -5,13 +5,13 @@ import streamlit as st
 
 
 def init_style():
-    return st.write(
+    return st.markdown(
         """
     <style>
     /* Side Bar */
-    .css-1outpf7 {
+    .sidebar .sidebar-content {
         background-color:rgb(254 244 219);
-        width:30rem;
+        width:10rem;
         padding:10px 10px 10px 10px;
     }
     /* Main Panel*/
@@ -61,6 +61,9 @@ def init_style():
         border-radius: 50px; 
     }
 
+    tbody th {display:none}
+    .blank {display:none}
+
     </style>
 """,
         unsafe_allow_html=True,