Warn users w FE experimental feature (#147)

autogluon · Nov 18, 2024 · bc6d15e · bc6d15e
1 parent bbdf2e5
commit bc6d15e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/autogluon/assistant/__init__.py b/src/autogluon/assistant/__init__.py
@@ -31,6 +31,7 @@
 __all__ = ["TabularPredictionAssistant", "TabularPredictionTask"]
 
 logging.basicConfig(level=logging.INFO)
+logging.getLogger("httpx").setLevel(logging.ERROR)
 
 
 def get_task(path: Path) -> TabularPredictionTask:

diff --git a/src/autogluon/assistant/ui/pages/task.py b/src/autogluon/assistant/ui/pages/task.py
@@ -125,12 +125,19 @@ def config_llm():
 
 def config_feature_generation():
     load_value("feature_generation")
-    st.checkbox(
+    checkbox = st.checkbox(
         "Feature Generation (Beta Feature)",
         key="_feature_generation",
         on_change=store_value,
         args=["feature_generation"],
     )
+    # Show warning if checkbox is checked
+    if checkbox:
+        st.warning(
+            "Feature Generation is an experimental feature which requires optional packages. Please install [requirements](https://github.com/autogluon/autogluon-assistant/blob/main/requirements.txt) with: "
+            "'pip install -r requirements.txt'",
+            icon="⚠️",
+        )
 
 
 def store_value_and_save_file(key):