KennethEnevoldsen commited on
Commit
b17c0b6
2 Parent(s): 4d1b40c 5c3749d

Merge pull request #28 from sanjeev-kumar-4/main

Browse files

Enable 'Embedding Dimensions' and 'Max Tokens' in all the tabs

Files changed (1) hide show
  1. refresh.py +1 -4
refresh.py CHANGED
@@ -603,7 +603,7 @@ def get_mteb_average(task_dict: dict) -> tuple[Any, dict]:
603
  for task_category, task_category_list in task_dict.items():
604
  DATA_TASKS[task_category] = add_rank(
605
  DATA_OVERALL[
606
- ["Model", "Model Size (Million Parameters)", "Memory Usage (GB, fp32)"] + task_category_list
607
  ]
608
  )
609
  DATA_TASKS[task_category] = DATA_TASKS[task_category][
@@ -670,9 +670,6 @@ def refresh_leaderboard() -> tuple[list, dict]:
670
  data_task_category = get_mteb_data(
671
  tasks=[task_category], datasets=task_category_list
672
  )
673
- data_task_category.drop(
674
- columns=["Embedding Dimensions", "Max Tokens"], inplace=True
675
- )
676
  boards_data[board]["data_tasks"][task_category] = data_task_category
677
  all_data_tasks.append(data_task_category)
678
  if board == "bright_long":
 
603
  for task_category, task_category_list in task_dict.items():
604
  DATA_TASKS[task_category] = add_rank(
605
  DATA_OVERALL[
606
+ ["Model", "Model Size (Million Parameters)", "Memory Usage (GB, fp32)", "Embedding Dimensions", "Max Tokens"] + task_category_list
607
  ]
608
  )
609
  DATA_TASKS[task_category] = DATA_TASKS[task_category][
 
670
  data_task_category = get_mteb_data(
671
  tasks=[task_category], datasets=task_category_list
672
  )
 
 
 
673
  boards_data[board]["data_tasks"][task_category] = data_task_category
674
  all_data_tasks.append(data_task_category)
675
  if board == "bright_long":