[Enhancement] version check codebase from openmmlab (#2006)

This commit is contained in:
Miao Zheng 2022-08-31 18:46:43 +08:00 committed by GitHub
parent ef111dff8b
commit 8ba80176a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 5 deletions

View File

@ -2,12 +2,14 @@
import warnings
import mmcv
import mmengine
from packaging.version import parse
from .version import __version__, version_info
MMCV_MIN = '2.0.0rc0'
MMCV_MAX = '2.0.0'
MMCV_MIN = '2.0.0rc1'
MMCV_MAX = '2.1.0'
MMENGINE_MIN = '0.5.0'
def digit_version(version_str: str, length: int = 4):
@ -59,4 +61,11 @@ assert (mmcv_min_version <= mmcv_version < mmcv_max_version), \
f'MMCV=={mmcv.__version__} is used but incompatible. ' \
f'Please install mmcv>={mmcv_min_version}, <{mmcv_max_version}.'
mmengine_min_version = digit_version(MMENGINE_MIN)
mmengine_version = digit_version(mmengine.__version__)
assert (mmengine_min_version <= mmengine_version), \
f'MMEngine=={mmengine.__version__} is used but incompatible. ' \
f'Please install mmengine>={mmengine_min_version}.'
__all__ = ['__version__', 'version_info', 'digit_version']

View File

@ -1,2 +1,3 @@
mmcls>=0.20.1
mmcv-full>=2.0.0,<=2.0.0rc0
mmcls>=1.0.0rc0
mmcv>=2.0.0rc1,<2.1.0
mmengine>=0.5

View File

@ -1,5 +1,5 @@
matplotlib
mmcls>=0.20.1
mmcls>=1.0.0rc0
numpy
packaging
prettytable