version.py 800 Bytes
Newer Older
Wenwei Zhang's avatar
Wenwei Zhang committed
1
2
# Copyright (c) Open-MMLab. All rights reserved.

Jingwei Zhang's avatar
Jingwei Zhang committed
3
__version__ = '1.1.0rc2'
Wenwei Zhang's avatar
Wenwei Zhang committed
4
5
6
7
short_version = __version__


def parse_version_info(version_str):
8
9
10
11
12
13
14
15
16
    """Parse a version string into a tuple.

    Args:
        version_str (str): The version string.

    Returns:
        tuple[int | str]: The version info, e.g., "1.3.0" is parsed into
            (1, 3, 0), and "2.0.0rc1" is parsed into (2, 0, 0, 'rc1').
    """
Wenwei Zhang's avatar
Wenwei Zhang committed
17
18
19
20
21
22
23
24
25
26
27
28
    version_info = []
    for x in version_str.split('.'):
        if x.isdigit():
            version_info.append(int(x))
        elif x.find('rc') != -1:
            patch_version = x.split('rc')
            version_info.append(int(patch_version[0]))
            version_info.append(f'rc{patch_version[1]}')
    return tuple(version_info)


version_info = parse_version_info(__version__)