fix column selection bug

app.py

@@ -18,7 +18,7 @@ from src.about import (
 from src.display.css_html_js import custom_css
 from src.display.utils import (
     BENCHMARK_COLS,
-    COLS,
+    #COLS,
     COLS_NC,
     COLS_NR,
     COLS_LP,
@@ -70,6 +70,7 @@ def update_table(
 ):
     #filtered_df = filter_models(hidden_df, size_query, show_deleted)
     filtered_df = filter_queries(query, hidden_df)
+    print(columns)
     df = select_columns(filtered_df, columns)
     return df
 
@@ -83,9 +84,13 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
         "Model"
     ]
     # We use COLS to maintain sorting
+    #print(df)
+    #print(df.columns)
+    #print([c for c in df.columns if c in columns])
     filtered_df = df[
-        always_here_cols + [c for c in COLS if c in df.columns and c in columns] 
+        always_here_cols + [c for c in df.columns if c in columns]
     ]
+    #print(filtered_df)
     return filtered_df
 
 
@@ -135,6 +140,7 @@ with demo:
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🏅 Entity Classification Leaderboard", elem_id="llm-benchmark-tab-table", id=0):
+            global COLS
             COLS = COLS_NC
             AutoEvalColumn = AutoEvalColumn_NodeClassification
             original_df = get_leaderboard_df(EVAL_REQUESTS_PATH, "Node Classification")
@@ -165,8 +171,8 @@ with demo:
                             interactive=True,
                         )
                     
-            print(leaderboard_df)
-            print(fields(AutoEvalColumn))
+            #print(leaderboard_df)
+            #print(shown_columns.value)
             leaderboard_table = gr.components.Dataframe(
                 value=leaderboard_df[
                     [c.name for c in fields(AutoEvalColumn) if c.never_hidden]
@@ -239,8 +245,8 @@ with demo:
                             interactive=True,
                         )
                     
-            print(leaderboard_df)
-            print(fields(AutoEvalColumn))
+            #print(leaderboard_df)
+            #print(shown_columns)
             leaderboard_table = gr.components.Dataframe(
                 value=leaderboard_df[
                     [c.name for c in fields(AutoEvalColumn) if c.never_hidden]
@@ -312,8 +318,8 @@ with demo:
                             interactive=True,
                         )
                     
-            print(leaderboard_df)
-            print(fields(AutoEvalColumn))
+            #print(leaderboard_df)
+            #print(shown_columns)
             leaderboard_table = gr.components.Dataframe(
                 value=leaderboard_df[
                     [c.name for c in fields(AutoEvalColumn) if c.never_hidden]