File size: 873 Bytes
cb44915
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""
CLI entry point for running evaluation as a module.

Usage:
    python -m src --model ./my_model/final
    python -m src --model username/model-name
"""

import argparse
import sys

from .evaluate import evaluate_model


def main():
    parser = argparse.ArgumentParser(
        description="Evaluate a chess model",
        prog="python -m src",
    )
    parser.add_argument(
        "--model",
        "-m",
        type=str,
        required=True,
        help="Path to model directory or HuggingFace model ID",
    )
    parser.add_argument(
        "--quiet",
        "-q",
        action="store_true",
        help="Suppress progress output",
    )
    
    args = parser.parse_args()
    
    result = evaluate_model(args.model, verbose=not args.quiet)
    print()
    print(result.summary())
    
    return 0


if __name__ == "__main__":
    sys.exit(main())