clean up logging, use extra args

ssube · Jan 28, 2023 · 3359138 · 3359138
1 parent 4547bce
commit 3359138
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/api/logging.yaml b/api/logging.yaml
@@ -1,7 +1,7 @@
 version: 1
 formatters:
   simple:
-    format: '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    format: '[%(asctime)s] %(levelname)s: %(name)s: %(message)s'
 handlers:
   console:
     class: logging.StreamHandler

diff --git a/api/onnx_web/__init__.py b/api/onnx_web/__init__.py
@@ -1,3 +1,5 @@
+from . import logging
+
 from .chain import (
   correct_gfpgan,
   upscale_resrgan,

diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py
@@ -71,7 +71,7 @@ def __call__(self, ctx: ServerContext, params: ImageParams, source: Image.Image,
             kwargs = {**pipeline_kwargs, **kwargs}
 
             logger.info('running stage %s on result image with dimensions %sx%s, %s',
-                        name, image.width, image.height, kwargs)
+                        name, image.width, image.height, kwargs.keys())
 
             if image.width > stage_params.tile_size or image.height > stage_params.tile_size:
                 logger.info('source image larger than tile size of %s, tiling stage',