diff --git a/mmseg/__init__.py b/mmseg/__init__.py index 8ffaf446b..addfaeedd 100644 --- a/mmseg/__init__.py +++ b/mmseg/__init__.py @@ -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'] diff --git a/requirements/mminstall.txt b/requirements/mminstall.txt index 91571a9c2..a06084502 100644 --- a/requirements/mminstall.txt +++ b/requirements/mminstall.txt @@ -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 diff --git a/requirements/runtime.txt b/requirements/runtime.txt index 520408fe8..82b392e74 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -1,5 +1,5 @@ matplotlib -mmcls>=0.20.1 +mmcls>=1.0.0rc0 numpy packaging prettytable