Switch to unified view

a b/ViTPose/mmpose/__init__.py
1
# Copyright (c) OpenMMLab. All rights reserved.
2
import mmcv
3
4
from .version import __version__, short_version
5
6
7
def digit_version(version_str):
8
    digit_version = []
9
    for x in version_str.split('.'):
10
        if x.isdigit():
11
            digit_version.append(int(x))
12
        elif x.find('rc') != -1:
13
            patch_version = x.split('rc')
14
            digit_version.append(int(patch_version[0]) - 1)
15
            digit_version.append(int(patch_version[1]))
16
    return digit_version
17
18
19
mmcv_minimum_version = '1.3.8'
20
mmcv_maximum_version = '1.5.0'
21
mmcv_version = digit_version(mmcv.__version__)
22
23
24
assert (mmcv_version >= digit_version(mmcv_minimum_version)
25
        and mmcv_version <= digit_version(mmcv_maximum_version)), \
26
    f'MMCV=={mmcv.__version__} is used but incompatible. ' \
27
    f'Please install mmcv>={mmcv_minimum_version}, <={mmcv_maximum_version}.'
28
29
__all__ = ['__version__', 'short_version']