Spaces:
Running
Running
Commit
·
b63b8ed
1
Parent(s):
3f487cb
init
Browse files
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 =
|
| 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 |
|