Browse Source

chore: logging env variable

AlpinDale 1 year ago
parent
commit
a3cab09b69
1 changed files with 3 additions and 1 deletions
  1. 3 1
      aphrodite/common/logger.py

+ 3 - 1
aphrodite/common/logger.py

@@ -5,6 +5,8 @@ https://github.com/skypilot-org/skypilot/blob/master/sky/sky_logging.py
 
 import logging
 import sys
+import os
+
 import colorlog
 
 # pylint: disable=line-too-long
@@ -71,7 +73,7 @@ _setup_logger()
 
 def init_logger(name: str):
     logger = logging.getLogger(name)
-    logger.setLevel(logging.DEBUG)
+    logger.setLevel(os.getenv("LOG_LEVEL", "DEBUG"))
     logger.addHandler(_default_handler)
     logger.propagate = False
     return logger