Update app.py
Browse files
app.py
CHANGED
@@ -43,13 +43,13 @@ def get_leaderboard_for_task(task_pretty: str) -> gr.components.Dataframe:
|
|
43 |
datatype=get_types_per_task(TASKS_PRETTY_REVERSE[task_pretty]),
|
44 |
)
|
45 |
|
46 |
-
code_completion_dataset_names = get_results_for_task(TASKS_PRETTY['project_code_completion'])['
|
47 |
|
48 |
def get_leaderboard_for_completion_task(dataset_name: str | None):
|
49 |
df = get_results_for_task(TASKS_PRETTY['project_code_completion'])
|
50 |
if dataset_name is None:
|
51 |
dataset_name = code_completion_dataset_names[0]
|
52 |
-
filtered_df = df[df['Dataset']==dataset_name]
|
53 |
return gr.components.Dataframe(
|
54 |
value=filtered_df,
|
55 |
interactive=False,
|
|
|
43 |
datatype=get_types_per_task(TASKS_PRETTY_REVERSE[task_pretty]),
|
44 |
)
|
45 |
|
46 |
+
code_completion_dataset_names = get_results_for_task(TASKS_PRETTY['project_code_completion'])['Dataset Name'].unique().tolist()
|
47 |
|
48 |
def get_leaderboard_for_completion_task(dataset_name: str | None):
|
49 |
df = get_results_for_task(TASKS_PRETTY['project_code_completion'])
|
50 |
if dataset_name is None:
|
51 |
dataset_name = code_completion_dataset_names[0]
|
52 |
+
filtered_df = df[df['Dataset Name']==dataset_name]
|
53 |
return gr.components.Dataframe(
|
54 |
value=filtered_df,
|
55 |
interactive=False,
|