fix(logging): update how get_logger sets logging level to support int (#26)

* fix(logging): update how get_logger sets logging level to support int based log level env variable.

* chore: bump up version for release.

Author: zhaozuy

python/model_hosting_container_standards/__init__.py

@@ -5,4 +5,4 @@
 - FastAPI: from .common.fastapi import EnvVars, ENV_CONFIG
 """
 
-__version__ = "0.1.8"
+__version__ = "0.1.9"

python/model_hosting_container_standards/logging_config.py

@@ -19,9 +19,10 @@ def get_logger(name: str = "model_hosting_container_standards") -> logging.Logge
     # Only configure if not already configured
     if not logger.handlers:
         # Get log level from environment, default to ERROR (effectively disabled)
+        # Convert level to uppercase
         level = os.getenv(
             "SAGEMAKER_CONTAINER_LOG_LEVEL", os.getenv("LOG_LEVEL", "ERROR")
-        )
+        ).upper()
 
         # Set up handler with consistent format
         handler = logging.StreamHandler(sys.stdout)
@@ -30,7 +31,11 @@ def get_logger(name: str = "model_hosting_container_standards") -> logging.Logge
         )
         handler.setFormatter(formatter)
         logger.addHandler(handler)
-        logger.setLevel(getattr(logging, level.upper()))
+
+        # Convert numeric log level string to int so `setLevel` can recognize it
+        # Will raise error if the set log level is not registered in
+        # logging.getLevelNamesMapping()
+        logger.setLevel(int(level) if level.isdigit() else level.upper())
 
         # Prevent propagation to avoid duplicate logs
         logger.propagate = False

python/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "model-hosting-container-standards"
-version = "0.1.8"
+version = "0.1.9"
 description = "Python toolkit for standardized model hosting container implementations with Amazon SageMaker integration"
 authors = ["Amazon Web Services"]
 readme = "README.md"