future-xy commited on
Commit
a4a186c
1 Parent(s): b4759e0

set log level to warning

Browse files
Files changed (2) hide show
  1. backend-cli.py +13 -2
  2. src/backend/run_eval_suite.py +1 -0
backend-cli.py CHANGED
@@ -25,10 +25,21 @@ from typing import Optional
25
 
26
  import time
27
 
28
- import logging
29
  import pprint
30
- from lm_eval.utils import eval_logger
 
 
 
 
 
 
 
 
 
 
 
31
 
 
32
  eval_logger.setLevel(logging.WARNING)
33
 
34
 
 
25
 
26
  import time
27
 
 
28
  import pprint
29
+ import logging
30
+
31
+
32
+ # Configure the root logger
33
+ logging.basicConfig(
34
+ format="%(asctime)s,%(msecs)03d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s",
35
+ datefmt="%Y-%m-%d:%H:%M:%S",
36
+ level=logging.WARNING,
37
+ )
38
+
39
+ # Get the 'lm-eval' logger from the third-party library
40
+ eval_logger = logging.getLogger("lm-eval")
41
 
42
+ # Explicitly set the level for 'lm-eval' logger to WARNING
43
  eval_logger.setLevel(logging.WARNING)
44
 
45
 
src/backend/run_eval_suite.py CHANGED
@@ -59,6 +59,7 @@ def run_evaluation(
59
  limit=limit,
60
  write_out=True,
61
  task_manager=task_manager,
 
62
  )
63
 
64
  results["config"]["model_dtype"] = eval_request.precision
 
59
  limit=limit,
60
  write_out=True,
61
  task_manager=task_manager,
62
+ verbosity="WARNING",
63
  )
64
 
65
  results["config"]["model_dtype"] = eval_request.precision