refine log format

Author: nbcsm

tests/common.py

@@ -116,7 +116,7 @@ def get_test_config():
 
 def unittest_main():
     config = get_test_config()
-    logging.basicConfig(level=logging.WARNING)
+    logging.basicConfig(level=logging.WARNING, format=constants.LOG_FORMAT)
     with utils.set_log_level(logging.getLogger(), logging.INFO) as logger:
         logger.info(config)
     unittest.main()

tests/conftest.py

@@ -5,11 +5,11 @@
 
 import logging
 from common import get_test_config
-from tf2onnx import utils
+from tf2onnx import constants, utils
 
 
 def pytest_configure():
     config = get_test_config()
-    logging.basicConfig(level=logging.WARNING)
+    logging.basicConfig(level=logging.WARNING, format=constants.LOG_FORMAT)
     with utils.set_log_level(logging.getLogger(), logging.INFO) as logger:
         logger.info(config)

tests/run_pretrained_models.py

@@ -28,14 +28,12 @@
 import yaml
 
 import tf2onnx
-from tf2onnx import loader
-from tf2onnx import utils
-from tf2onnx import optimizer
+from tf2onnx import constants, loader, optimizer, utils
 from tf2onnx.tfonnx import process_tf_graph
 
 # pylint: disable=broad-except,logging-not-lazy,unused-argument,unnecessary-lambda
 
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(level=logging.INFO, format=constants.LOG_FORMAT)
 log = logging.getLogger("tf2onnx")
 
 TEMP_DIR = os.path.join(utils.get_temp_directory(), "run_pretrained")

tf2onnx/constants.py

@@ -32,3 +32,5 @@
 NHWC_TO_NCHW = [0, 3, 1, 2]
 HWCN_TO_NCHW = [3, 2, 0, 1]
 NCHW_TO_HWCN = [2, 3, 1, 0]
+
+LOG_FORMAT = "%(asctime)s - %(levelname)s - %(name)s: %(message)s"

tf2onnx/convert.py

@@ -80,7 +80,7 @@ def default_custom_op_handler(ctx, node, name, args):
 
 def main():
     args = get_args()
-    logging.basicConfig(level=logging.INFO)
+    logging.basicConfig(level=logging.INFO, format=constants.LOG_FORMAT)
 
     # override unknown dimensions from -1 to 1 (aka batchsize 1) since not every runtime does
     # support unknown dimensions.