precise removal verbosity
Browse files- 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")
|