hysts HF staff commited on
Commit
c0691b9
1 Parent(s): 99a4ea0
Files changed (1) hide show
  1. src/populate.py +1 -1
src/populate.py CHANGED
@@ -15,7 +15,6 @@ def get_leaderboard_df(contents_repo: str, cols: list, benchmark_cols: list) ->
15
  df = datasets.load_dataset(contents_repo, split="train").to_pandas()
16
  df["Model"] = df["model"].map(make_clickable_model)
17
  df["T"] = df["model_type"].map(lambda x: x.split(":")[0].strip())
18
- df["Type"] = df["model_type"].map(lambda x: x.split(":")[1].strip())
19
  df["Backend Library"] = "vllm"
20
  df = df.rename(columns={task.value.metric: task.value.col_name for task in Tasks})
21
  df = df.rename(
@@ -31,6 +30,7 @@ def get_leaderboard_df(contents_repo: str, cols: list, benchmark_cols: list) ->
31
  "add_special_tokens": "Add Special Tokens",
32
  "llm_jp_eval_version": "llm-jp-eval version",
33
  "vllm_version": "vllm version",
 
34
  "model": "model_name_for_query",
35
  }
36
  )
 
15
  df = datasets.load_dataset(contents_repo, split="train").to_pandas()
16
  df["Model"] = df["model"].map(make_clickable_model)
17
  df["T"] = df["model_type"].map(lambda x: x.split(":")[0].strip())
 
18
  df["Backend Library"] = "vllm"
19
  df = df.rename(columns={task.value.metric: task.value.col_name for task in Tasks})
20
  df = df.rename(
 
30
  "add_special_tokens": "Add Special Tokens",
31
  "llm_jp_eval_version": "llm-jp-eval version",
32
  "vllm_version": "vllm version",
33
+ "model_type": "Type",
34
  "model": "model_name_for_query",
35
  }
36
  )