From 21fa864317e3a317a26d0f9da09c57ee0dccc06b Mon Sep 17 00:00:00 2001
From: Domenic Barbuzzi <domenic@neuralmagic.com>
Date: Wed, 29 May 2024 16:54:37 +0000
Subject: [PATCH] Make copy of logging utils

---
 .../benchmarks/run_benchmark_serving.py       |  2 +-
 neuralmagic/tools/logging.py                  | 33 +++++++++++++++++++
 2 files changed, 34 insertions(+), 1 deletion(-)
 create mode 100644 neuralmagic/tools/logging.py

diff --git a/neuralmagic/benchmarks/run_benchmark_serving.py b/neuralmagic/benchmarks/run_benchmark_serving.py
index 6a7b8743f7b6f..826949491a06a 100644
--- a/neuralmagic/benchmarks/run_benchmark_serving.py
+++ b/neuralmagic/benchmarks/run_benchmark_serving.py
@@ -11,8 +11,8 @@
 
 import requests
 
-from ...tests.utils.logging import log_banner, make_logger
 from ..tools.call_cmd import call_cmd
+from ..tools.logging import log_banner, make_logger
 from .common import (benchmark_configs, download_model,
                      max_model_length_from_model_id, script_args_to_cla)
 from .scripts.common import num_available_gpus, warmup_server
diff --git a/neuralmagic/tools/logging.py b/neuralmagic/tools/logging.py
new file mode 100644
index 0000000000000..a63463db61252
--- /dev/null
+++ b/neuralmagic/tools/logging.py
@@ -0,0 +1,33 @@
+import logging
+
+
+def make_logger(name: str) -> logging.Logger:
+    """Create a base logger"""
+
+    logger = logging.getLogger(name)
+    logger.setLevel(logging.DEBUG)
+    stream_handler = logging.StreamHandler()
+    formatter = logging.Formatter(
+        "%(asctime)s - %(levelname)s - %(name)s - %(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+    )
+    stream_handler.setFormatter(formatter)
+    logger.addHandler(stream_handler)
+    return logger
+
+
+def log_banner(logger: logging.Logger,
+               label: str,
+               body: str,
+               level: int = logging.INFO):
+    """
+    Log a message in the "banner"-style format.
+
+    :param logger: Instance of "logging.Logger" to use
+    :param label: Label for the top of the banner
+    :param body: Body content inside the banner
+    :param level: Logging level to use (default: INFO)
+    """
+
+    banner = f"==== {label} ====\n{body}\n===="
+    logger.log(level, "\n%s", banner)