Skip to content

Commit 75329d1

Browse files
committed
get_verbosity_level relative verbosity
1 parent 2b3c23d commit 75329d1

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tf2onnx/logging.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -47,15 +47,15 @@ def basicConfig(**kwargs): # pylint: disable=invalid-name, function-redefined
4747
set_tf_verbosity(_logging.getLogger().getEffectiveLevel())
4848

4949

50-
_VERBOSITY_TO_LEVEL = [INFO, VERBOSE, DEBUG]
50+
_LOG_LEVELS = [FATAL, ERROR, WARNING, INFO, VERBOSE, DEBUG]
5151

5252

53-
def get_verbosity_level(verbosity, default_level=INFO):
53+
def get_verbosity_level(verbosity, base_level=INFO):
5454
""" If verbosity is specified, return corresponding level, otherwise, return default_level. """
5555
if verbosity is None:
56-
return default_level
57-
verbosity = min(max(0, verbosity), len(_VERBOSITY_TO_LEVEL) - 1)
58-
return _VERBOSITY_TO_LEVEL[verbosity]
56+
return base_level
57+
verbosity = min(max(0, verbosity) + _LOG_LEVELS.index(base_level), len(_LOG_LEVELS) - 1)
58+
return _LOG_LEVELS[verbosity]
5959

6060

6161
def set_level(level):

0 commit comments

Comments
 (0)