36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
import argparse
|
|
|
|
from evalscope import __version__
|
|
from evalscope.cli.benchmark_info import BenchmarkInfoCMD
|
|
from evalscope.cli.start_app import StartAppCMD
|
|
from evalscope.cli.start_eval import EvalCMD
|
|
from evalscope.cli.start_perf import PerfBenchCMD
|
|
from evalscope.cli.start_service import ServiceCMD
|
|
|
|
|
|
def run_cmd():
|
|
parser = argparse.ArgumentParser('EvalScope Command Line tool', usage='evalscope <command> [<args>]')
|
|
parser.add_argument('-v', '--version', action='version', version=f'evalscope {__version__}')
|
|
subparsers = parser.add_subparsers(help='EvalScope command line helper.')
|
|
|
|
PerfBenchCMD.define_args(subparsers)
|
|
EvalCMD.define_args(subparsers)
|
|
StartAppCMD.define_args(subparsers)
|
|
ServiceCMD.define_args(subparsers)
|
|
BenchmarkInfoCMD.define_args(subparsers)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not hasattr(args, 'func'):
|
|
parser.print_help()
|
|
exit(1)
|
|
|
|
cmd = args.func(args)
|
|
cmd.execute()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run_cmd()
|