linter fixes

This commit is contained in:
erogol 2020-02-14 18:00:15 +01:00
parent 0c7c34c12c
commit ecf84fa4ad
1 changed files with 3 additions and 11 deletions

View File

@ -402,8 +402,8 @@ def _check_argument(name, c, enum_list=None, max_val=None, min_val=None, restric
if enum_list: if enum_list:
assert c[name].lower() in enum_list, f' [!] {name} is not a valid value' assert c[name].lower() in enum_list, f' [!] {name} is not a valid value'
if val_type: if val_type:
assert type(c[name]) is val_type or c[name] is None, f' [!] {name} has wrong type - {type(c[name])} vs {val_type}' assert isinstance(c[name], val_type) or c[name] is None, f' [!] {name} has wrong type - {type(c[name])} vs {val_type}'
def check_config(c): def check_config(c):
_check_argument('model', c, enum_list=['tacotron', 'tacotron2'], restricted=True, val_type=str) _check_argument('model', c, enum_list=['tacotron', 'tacotron2'], restricted=True, val_type=str)
@ -507,12 +507,4 @@ def check_config(c):
_check_argument('name', dataset_entry, restricted=True, val_type=str) _check_argument('name', dataset_entry, restricted=True, val_type=str)
_check_argument('path', dataset_entry, restricted=True, val_type=str) _check_argument('path', dataset_entry, restricted=True, val_type=str)
_check_argument('meta_file_train', dataset_entry, restricted=True, val_type=str) _check_argument('meta_file_train', dataset_entry, restricted=True, val_type=str)
_check_argument('meta_file_val', dataset_entry, restricted=True, val_type=str) _check_argument('meta_file_val', dataset_entry, restricted=True, val_type=str)