aboutsummaryrefslogtreecommitdiffstats
path: root/snips_inference_agl/cli/versions.py
blob: d7922efcfc66fe5ef25eb5217d1cdb7da66bf13e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from __future__ import print_function


def add_version_parser(subparsers, formatter_class):
    from snips_inference_agl.__about__ import __version__
    subparser = subparsers.add_parser(
        "version", formatter_class=formatter_class,
        help="Print the package version")
    subparser.set_defaults(func=lambda _: print(__version__))
    return subparser


def add_model_version_parser(subparsers, formatter_class):
    from snips_inference_agl.__about__ import __model_version__
    subparser = subparsers.add_parser(
        "model-version", formatter_class=formatter_class,
        help="Print the model version")
    subparser.set_defaults(func=lambda _: print(__model_version__))
    return subparser