Support logger_formatter without automl dependencies (#1356)

microsoft · Sep 21, 2024 · 49ba962 · 49ba962
1 parent 8e171bc
commit 49ba962
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/flaml/automl/__init__.py b/flaml/automl/__init__.py
@@ -1,5 +1,9 @@
-from flaml.automl.automl import AutoML, size
 from flaml.automl.logger import logger_formatter
-from flaml.automl.state import AutoMLState, SearchState
 
-__all__ = ["AutoML", "AutoMLState", "SearchState", "logger_formatter", "size"]
+try:
+    from flaml.automl.automl import AutoML, size
+    from flaml.automl.state import AutoMLState, SearchState
+
+    __all__ = ["AutoML", "AutoMLState", "SearchState", "logger_formatter", "size"]
+except ImportError:
+    __all__ = ["logger_formatter"]