fix: Add formatter to logger with a timestamp. (#95)

fix: Add formatter to logger with a timestamp.

Author: davidthomas426

src/sagemaker_inference/logging.py

@@ -16,6 +16,16 @@
 import logging
 
 
+def configure_logger():
+    """Add a handler to the library's logger with a formatter that
+    includes a timestamp along with the message.
+    """
+    handler = logging.StreamHandler()
+    formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+    handler.setFormatter(formatter)
+    get_logger().addHandler(handler)
+
+
 def get_logger():
     """Return a logger with the name "sagemaker-inference",
     creating it if necessary.

src/sagemaker_inference/model_server.py

@@ -27,6 +27,7 @@
 from sagemaker_inference import default_handler_service, environment, logging, utils
 from sagemaker_inference.environment import code_dir
 
+logging.configure_logger()
 logger = logging.get_logger()
 
 MMS_CONFIG_FILE = os.path.join("/etc", "sagemaker-mms.properties")