Skip to content

Commit f0d5b4b

Browse files
committed
add verbose level name, rename get_adjusted_level
1 parent 386b69a commit f0d5b4b

File tree

4 files changed

+8
-6
lines changed

4 files changed

+8
-6
lines changed

tests/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def load():
9595
config.backend = args.backend
9696
config.opset = args.opset
9797
config.target = args.target.split(',')
98-
config.log_level = logging.get_adjusted_level(args.verbose, config.log_level)
98+
config.log_level = logging.get_verbosity_level(args.verbose, config.log_level)
9999
config.is_debug_mode = args.debug
100100
if args.temp_dir:
101101
config.temp_dir = args.temp_dir

tests/run_pretrained_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -371,7 +371,7 @@ def tests_from_yaml(fname):
371371

372372
def main():
373373
args = get_args()
374-
logging.basicConfig(level=logging.get_adjusted_level(args.verbose))
374+
logging.basicConfig(level=logging.get_verbosity_level(args.verbose))
375375

376376
Test.cache_dir = args.cache
377377
Test.target = args.target

tf2onnx/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def default_custom_op_handler(ctx, node, name, args):
8080

8181
def main():
8282
args = get_args()
83-
logging.basicConfig(level=logging.get_adjusted_level(args.verbose))
83+
logging.basicConfig(level=logging.get_verbosity_level(args.verbose))
8484

8585
# override unknown dimensions from -1 to 1 (aka batchsize 1) since not every runtime does
8686
# support unknown dimensions.

tf2onnx/logging.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717

1818
VERBOSE = 15
1919

20+
_logging.addLevelName(VERBOSE, "VERBOSE")
21+
2022

2123
def _verbose(self, message, *args, **kwargs):
2224
if self.isEnabledFor(VERBOSE):
@@ -51,9 +53,9 @@ def basicConfig(**kwargs): # pylint: disable=invalid-name, function-redefined
5153
}
5254

5355

54-
def get_adjusted_level(verbosity, raw_level=INFO):
55-
""" If verbosity is specified, return corresponding level, otherwise, return raw_level. """
56-
return _VERBOSITY_TO_LEVEL.get(verbosity, raw_level)
56+
def get_verbosity_level(verbosity, default_level=INFO):
57+
""" If verbosity is specified, return corresponding level, otherwise, return default_level. """
58+
return _VERBOSITY_TO_LEVEL.get(verbosity, default_level)
5759

5860

5961
def set_level(level):

0 commit comments

Comments
 (0)