smothiki commited on
Commit
9803cb7
·
1 Parent(s): dfb2bc4

Update src/tools/collections.py

Browse files
Files changed (1) hide show
  1. src/tools/collections.py +2 -2
src/tools/collections.py CHANGED
@@ -6,7 +6,7 @@ from huggingface_hub.utils._errors import HfHubHTTPError
6
  from pandas import DataFrame
7
 
8
  from src.display.utils import AutoEvalColumn, ModelType
9
- from src.envs import H4_TOKEN, PATH_TO_COLLECTION
10
 
11
  # Specific intervals for the collections
12
  intervals = {
@@ -23,7 +23,7 @@ def update_collections(df: DataFrame):
23
  """This function updates the Open LLM Leaderboard model collection with the latest best models for
24
  each size category and type.
25
  """
26
- collection = get_collection(collection_slug=PATH_TO_COLLECTION, token=H4_TOKEN)
27
  params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
28
 
29
  cur_best_models = []
 
6
  from pandas import DataFrame
7
 
8
  from src.display.utils import AutoEvalColumn, ModelType
9
+ from src.envs import HF_TOKEN, PATH_TO_COLLECTION
10
 
11
  # Specific intervals for the collections
12
  intervals = {
 
23
  """This function updates the Open LLM Leaderboard model collection with the latest best models for
24
  each size category and type.
25
  """
26
+ collection = get_collection(collection_slug=PATH_TO_COLLECTION, token=HF_TOKEN)
27
  params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
28
 
29
  cur_best_models = []