__main__.py 296 Bytes
Newer Older
Baber's avatar
nit  
Baber committed
1
from lm_eval._cli.harness import HarnessCLI
Baber's avatar
Baber committed
2
from lm_eval.utils import setup_logging
Jason Phang's avatar
Jason Phang committed
3

Fabrizio Milo's avatar
Fabrizio Milo committed
4

Baber's avatar
Baber committed
5
def cli_evaluate() -> None:
Baber's avatar
nit  
Baber committed
6
    """Main CLI entry point."""
Baber's avatar
Baber committed
7
    setup_logging()
Baber's avatar
nit  
Baber committed
8
    parser = HarnessCLI()
Baber's avatar
Baber committed
9
10
    args = parser.parse_args()
    parser.execute(args)
11

12

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