Linker1907 commited on
Commit
b63b8ed
·
1 Parent(s): 3f487cb
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -30,7 +30,7 @@ TASK_DIRS = [
30
  os.path.join(REPO_ROOT, "src", "lighteval", "tasks", "tasks"),
31
  os.path.join(REPO_ROOT, "src", "lighteval", "tasks", "multilingual", "tasks"),
32
  ]
33
- CACHE_PATH = os.path.join(REPO_ROOT, "benchmark_finder", "tasks_index.json")
34
 
35
 
36
  @dataclass
@@ -180,9 +180,12 @@ def build_and_cache_index() -> Tuple[List[TaskDoc], List[str], List[str]]:
180
 
181
  _loaded = load_index(CACHE_PATH)
182
  if _loaded is None:
 
183
  ALL_TASKS, ALL_LANGS, ALL_TAGS = build_and_cache_index()
184
  else:
 
185
  ALL_TASKS, ALL_LANGS, ALL_TAGS = _loaded
 
186
  TOP_LANGS = ALL_LANGS[:5]
187
 
188
 
 
30
  os.path.join(REPO_ROOT, "src", "lighteval", "tasks", "tasks"),
31
  os.path.join(REPO_ROOT, "src", "lighteval", "tasks", "multilingual", "tasks"),
32
  ]
33
+ CACHE_PATH = "tasks_index.json"
34
 
35
 
36
  @dataclass
 
180
 
181
  _loaded = load_index(CACHE_PATH)
182
  if _loaded is None:
183
+ print("Building and caching index...")
184
  ALL_TASKS, ALL_LANGS, ALL_TAGS = build_and_cache_index()
185
  else:
186
+ print("Loading index from cache...")
187
  ALL_TASKS, ALL_LANGS, ALL_TAGS = _loaded
188
+ print(f"Loaded {len(ALL_TASKS)} tasks from cache")
189
  TOP_LANGS = ALL_LANGS[:5]
190
 
191