瀏覽代碼

chore: logging env variable

AlpinDale 1 年之前
父節點
當前提交
a3cab09b69
共有 1 個文件被更改,包括 3 次插入1 次删除
  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