Fix eval_logger import for mmlu/_generate_configs.py (#1593)

* Fix eval_logger import for mmlu/_generate_configs.py * linter --------- Co-authored-by: Hailey Schoelkopf <[email protected]>
EleutherAI · Mar 18, 2024 · 4600d6b · 4600d6b
1 parent 5627e81
commit 4600d6b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lm_eval/tasks/mmlu/_generate_configs.py b/lm_eval/tasks/mmlu/_generate_configs.py
@@ -2,12 +2,14 @@
 Take in a YAML, and output all "other" splits with this YAML
 """
 import argparse
+import logging
 import os
 
 import yaml
 from tqdm import tqdm
 
-from lm_eval.logger import eval_logger
+
+eval_logger = logging.getLogger("lm-eval")
 
 
 SUBJECTS = {