mirrors/mim
1
0
Fork 0
mirror of https://github.com/open-mmlab/mim.git synced 2025-06-03 14:59:11 +08:00
Code Issues Projects Releases Wiki Activity

[Enhancement] Add highlighted display when exception is raised (#19)

Browse Source
This commit is contained in:
Zaida Zhou 2021-05-24 22:24:07 +08:00 committed by GitHub
parent 11adbd231c
commit 855439eb2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mim/commands/download.py
View File

@ -61,12 +61,14 @@ def download(package: str,
package, version = split_package_version(package) package, version = split_package_version(package)
if version: if version:
raise ValueError('version is not allowed, please type ' raise ValueError(
'"mim download -h" to show the correct way.') highlighted_error('version is not allowed, please type '
'"mim download -h" to show the correct way.'))
if not is_installed(package): if not is_installed(package):
raise RuntimeError(f'{package} is not installed. Please install it ' raise RuntimeError(
'first.') highlighted_error(f'{package} is not installed. Please install it '
'first.'))
checkpoints = [] checkpoints = []
model_info = get_model_info( model_info = get_model_info(
@ -74,8 +76,9 @@ def download(package: str,
valid_configs = model_info.keys() valid_configs = model_info.keys()
invalid_configs = set(configs) - set(valid_configs) invalid_configs = set(configs) - set(valid_configs)
if invalid_configs: if invalid_configs:
raise ValueError(f'Expected configs: {valid_configs}, but got ' raise ValueError(
f'{invalid_configs}') highlighted_error(f'Expected configs: {valid_configs}, but got '
f'{invalid_configs}'))
from mmcv import Config from mmcv import Config