Skip to content

Commit e83a31d

Browse files
refactor: Utilise mapping
1 parent dc4fca2 commit e83a31d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/thread/utils/config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,12 @@ def __init__(self, level: VerbosityLevel) -> None:
3535
ValueError: If the level is not of a valid type.
3636
"""
3737
if isinstance(level, str):
38-
if level not in ['quiet', 'normal', 'verbose']:
38+
if level not in VerbosityMapping.keys():
3939
raise ValueError('Invalid verbosity level')
4040
self.level_string = level
4141
self.level_number = VerbosityMapping[level]
4242
elif isinstance(level, int):
43-
if level not in [0, 1, 2]:
43+
if level not in VerbosityMapping.values():
4444
raise ValueError('Invalid verbosity level')
4545
self.level_number = level
4646
self.level_string = list(VerbosityMapping.keys())[level]

0 commit comments

Comments
 (0)