no cache debug logging when not used

EleutherAI · Jan 4, 2025 · 0d42931 · 0d42931
1 parent 45006a2
commit 0d42931
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lm_eval/api/task.py b/lm_eval/api/task.py
@@ -398,7 +398,7 @@ def build_all_requests(
         )
         cache_key += f"-tokenizer{tokenizer_name}"
 
-        cached_instances = load_from_cache(file_name=cache_key)
+        cached_instances = load_from_cache(file_name=cache_key, cache=cache_requests)
 
         if cache_requests and cached_instances and not rewrite_requests_cache:
             cached_instances = cached_instances[:limit]

diff --git a/lm_eval/caching/cache.py b/lm_eval/caching/cache.py
@@ -21,7 +21,9 @@
 FILE_SUFFIX = f".{HASH_PREFIX}.pickle"
 
 
-def load_from_cache(file_name):
+def load_from_cache(file_name: str, cache: bool = False):
+    if not cache:
+        return
     try:
         path = f"{PATH}/{file_name}{FILE_SUFFIX}"