Skip to content

Commit b34102e

Browse files
committed
fix get_verbosity_level range
1 parent 0390798 commit b34102e

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 = {
51-
1: VERBOSE,
52-
2: DEBUG
53-
}
50+
_VERBOSITY_TO_LEVEL = [INFO, VERBOSE, DEBUG]
5451

5552

5653
def get_verbosity_level(verbosity, default_level=INFO):
5754
""" If verbosity is specified, return corresponding level, otherwise, return default_level. """
58-
return _VERBOSITY_TO_LEVEL.get(verbosity, default_level)
55+
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]
5959

6060

6161
def set_level(level):

0 commit comments

Comments
 (0)