Merge branch 'main' into thinkall-patch-1

microsoft · Nov 12, 2024 · 8b64c83 · 8b64c83
2 parents b82b0a2 + 07f4413
commit 8b64c83
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/flaml/__init__.py b/flaml/__init__.py
@@ -1,4 +1,5 @@
 import logging
+import warnings
 
 try:
     from flaml.automl import AutoML, logger_formatter
@@ -12,7 +13,8 @@
 
 # Set the root logger.
 logger = logging.getLogger(__name__)
-logger.setLevel(logging.INFO)
+if logger.level == logging.NOTSET:
+    logger.setLevel(logging.INFO)
 
 if not has_automl:
-    logger.warning("flaml.automl is not available. Please install flaml[automl] to enable AutoML functionalities.")
+    warnings.warn("flaml.automl is not available. Please install flaml[automl] to enable AutoML functionalities.")