Charlie81 commited on
Commit
181c6a9
·
1 Parent(s): aaada01

precise removal verbosity

Browse files
Files changed (1) hide show
  1. scripts/eval.py +7 -1
scripts/eval.py CHANGED
@@ -168,6 +168,13 @@ Examples:
168
  action="store_true",
169
  help="Trust remote code when loading model"
170
  )
 
 
 
 
 
 
 
171
 
172
  return parser.parse_args()
173
 
@@ -372,7 +379,6 @@ def run_evaluation(args) -> Dict[str, Any]:
372
  num_fewshot=args.num_fewshot,
373
  limit=args.limit,
374
  write_out=args.write_out,
375
- verbosity=args.verbosity,
376
  )
377
 
378
  logger.info("Evaluation completed successfully")
 
168
  action="store_true",
169
  help="Trust remote code when loading model"
170
  )
171
+ parser.add_argument(
172
+ "--verbosity",
173
+ type=str,
174
+ default="INFO",
175
+ choices=["DEBUG", "INFO", "WARNING", "ERROR"],
176
+ help="Logging verbosity level"
177
+ )
178
 
179
  return parser.parse_args()
180
 
 
379
  num_fewshot=args.num_fewshot,
380
  limit=args.limit,
381
  write_out=args.write_out,
 
382
  )
383
 
384
  logger.info("Evaluation completed successfully")