Skip to content

Commit 4600d6b

Browse files
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 <65563625+haileyschoelkopf@users.noreply.github.com>
1 parent 5627e81 commit 4600d6b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

lm_eval/tasks/mmlu/_generate_configs.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,14 @@
22
Take in a YAML, and output all "other" splits with this YAML
33
"""
44
import argparse
5+
import logging
56
import os
67

78
import yaml
89
from tqdm import tqdm
910

10-
from lm_eval.logger import eval_logger
11+
12+
eval_logger = logging.getLogger("lm-eval")
1113

1214

1315
SUBJECTS = {

0 commit comments

Comments
 (0)