__main__.py 485 Bytes
Newer Older
haileyschoelkopf's avatar
haileyschoelkopf committed
1
from typing import Union
Baber's avatar
Baber committed
2
import argparse
Leo Gao's avatar
Leo Gao committed
3

Baber's avatar
Baber committed
4
from lm_eval._cli import CLIParser
Jason Phang's avatar
Jason Phang committed
5

Fabrizio Milo's avatar
Fabrizio Milo committed
6

haileyschoelkopf's avatar
haileyschoelkopf committed
7
def cli_evaluate(args: Union[argparse.Namespace, None] = None) -> None:
Baber's avatar
Baber committed
8
9
    """Main CLI entry point with subcommand and legacy support."""
    parser = CLIParser()
KonradSzafer's avatar
KonradSzafer committed
10

Baber's avatar
Baber committed
11
12
13
    if args is None:
        # Parse from command line
        parser.execute()
Jason Phang's avatar
Jason Phang committed
14
    else:
Baber's avatar
Baber committed
15
16
        # External call with pre-parsed args - use legacy mode
        parser._handle_legacy_mode(args)
17

18

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