blob: bcd17c3312d4531c4718c44750eddd51a15cbcb7 (
plain)
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
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Module to mange the CLI interface."""
import click
from aiet import __version__
from aiet.cli.application import application_cmd
from aiet.cli.completion import completion_cmd
from aiet.cli.system import system_cmd
from aiet.cli.tool import tool_cmd
from aiet.utils.helpers import set_verbosity
@click.group()
@click.version_option(__version__)
@click.option(
"-v", "--verbose", default=0, count=True, callback=set_verbosity, expose_value=False
)
@click.pass_context
def cli(ctx: click.Context) -> None: # pylint: disable=unused-argument
"""AIET: AI Evaluation Toolkit."""
# Unused arguments must be present here in definition to pass click context.
cli.add_command(application_cmd)
cli.add_command(system_cmd)
cli.add_command(tool_cmd)
cli.add_command(completion_cmd)
|
