dropdown

app.py

@@ -27,7 +27,11 @@ logger = setup_logger(__name__)
 pp = pprint.PrettyPrinter(width=80)
 
 
-ALL_ENV_IDS = list(gym.registry.keys())
+ALL_ENV_IDS = [
+    "CartPole-v1",
+    "MountainCar-v0",
+    "Acrobot-v1",
+]
 
 
 def model_hyperlink(link, model_id):
@@ -221,9 +225,9 @@ The Open RL Leaderboard is a community-driven benchmark for reinforcement learni
 """
 
 
-def select_column(column_names, data):
-    column_names = [col for col in column_names if col in data.columns]
-    column_names = ["Agent"] + column_names  # add model name column
+def select_column(column_name: str, data: pd.DataFrame):
+    # column_names = [col for col in column_names if col in data.columns]
+    column_names = ["Agent"] + [column_name]  # add model name column
     df = data[column_names]
 
     def check_row(row):
@@ -231,6 +235,7 @@ def select_column(column_names, data):
 
     mask = df.apply(check_row, axis=1)
     df = df[mask]
+    df = df.sort_values(by=column_name, ascending=False)
     return df
 
 
@@ -239,18 +244,18 @@ with gr.Blocks(js=dark_mode_gradio_js) as demo:
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🏅 Leaderboard", elem_id="llm-benchmark-tab-table", id=0):
-            hidden_df = gr.components.Dataframe(get_leaderboard_df, visible=False, every=60)  # hidden dataframe
+            hidden_df = gr.components.Dataframe(get_leaderboard_df, visible=False, every=5 * 60)  # hidden dataframe
 
-            env_checkboxes = gr.components.CheckboxGroup(
+            env_selector = gr.components.Dropdown(
                 label="Environments",
                 choices=ALL_ENV_IDS,
-                value=[ALL_ENV_IDS[0]],
-                interactive=True,
+                value=ALL_ENV_IDS[0],
+                # interactive=True,
             )
-            leaderboard = gr.components.Dataframe(select_column([ALL_ENV_IDS[0]], get_leaderboard_df()))
+            leaderboard = gr.components.Dataframe(select_column(ALL_ENV_IDS[0], get_leaderboard_df()))
 
             # Events
-            env_checkboxes.change(select_column, [env_checkboxes, hidden_df], leaderboard)
+            env_selector.change(select_column, [env_selector, hidden_df], leaderboard)
             # Update hidden dataframe
             # hidden_df.change(get_leaderboard_df, [], hidden_df, every=10)
 
@@ -259,7 +264,7 @@ with gr.Blocks(js=dark_mode_gradio_js) as demo:
 
 
 scheduler = BackgroundScheduler()
-scheduler.add_job(func=backend_routine, trigger="interval", seconds=60)
+scheduler.add_job(func=backend_routine, trigger="interval", seconds=5 * 60)
 scheduler.start()