__main__.py 360 Bytes
Newer Older
Baber's avatar
Baber committed
1
2
from rich.traceback import install

Baber's avatar
Baber committed
3
from lm_eval._cli.harness import HarnessCLI
Baber's avatar
Baber committed
4
from lm_eval.utils import setup_logging
Leo Gao's avatar
Leo Gao committed
5

Jason Phang's avatar
Jason Phang committed
6

Baber's avatar
Baber committed
7
8
9
install(show_locals=True)


Baber's avatar
Baber committed
10
def cli_evaluate() -> None:
Baber's avatar
Baber committed
11
    """Main CLI entry point."""
Baber's avatar
Baber committed
12
    setup_logging()
Baber's avatar
Baber committed
13
    parser = HarnessCLI()
Baber's avatar
Baber committed
14
15
    args = parser.parse_args()
    parser.execute(args)
16

17

Jason Phang's avatar
Jason Phang committed
18
if __name__ == "__main__":
Baber's avatar
Baber committed
19
    cli_evaluate()